Update ch13_part3_lightning.ipynb

rasbt / machine-learning-book

Code Repository for Machine Learning with PyTorch and Scikit-Learn

https://sebastianraschka.com/books/#machine-learning-with-pytorch-and-scikit-learn

MIT License

3.63k stars 1.31k forks source link

Update ch13_part3_lightning.ipynb #154

Open akshayvkt opened 1 year ago

akshayvkt commented 1 year ago

Updated the deprecated modules (training_epoch_end and validation_epoch_end) to the new modules (on_train_epoch_end and on_validation_epoch_end).
Removed 'outs argument from both functions because the model fails to train with the outs argument in the function.

The model now successfully trains with these changes. I'm certain that change #1 is needed, I'm not sure about change #2 but updating that let me train the model and not error out.

review-notebook-app[bot] commented 1 year ago

Check out this pull request on

See visual diffs & provide feedback on Jupyter Notebooks.