Model class methods need to be overrided caused by Pytorch Lightning
Tony-Tseng opened this issue · 2 comments
Tony-Tseng commented
Thanks for your work.
I'm trying to reproduce the training result for super-resolution track 1.
However, after installing the environment by install.yml, I cannot run the training code by
python BIPNet_Track_1_training.py
.
The error appears as follows.
"No `training_step()` method defined. Lightning `Trainer` expects as minimum a"
pytorch_lightning.utilities.exceptions.MisconfigurationException: No `training_step()` method defined. Lightning `Trainer` expects as minimum a `training_step()`, `train_dataloader()` and `configure_optimizers()` to be defined.
I'm fixing it by adding training_step, and configure_optimizers functions in BIPNet and setting Adam optimizer with lr 1e-4 and Cosine annealing scheduler as the paper states. Is it the correct way to reproduce the result?
akshaydudhane16 commented
Issue is fixed by updating Network.py file.
Tony-Tseng commented
The code works! Thanks for the update.