Hi all, I implemented
configure_optimizers in my
pl.LightningModule.The model is training, however
training_epoch_end is never called. Am I missing something? I use the latest stable pytorch_lightning version 1.1.8.
Edit: This only happens when called from a Notebook in JupyterLab. When executing the script in the terminal, everything works fine.
Thanks a lot!