Skip to content

PyTorch Lightning Optimizer_Step() prevents training_step() from running #11358

Discussion options

You must be logged in to vote

hey @tsuijenk

optimizer _closure must be passed in optimizer.step() since it runs training_step and backward call. You can check the docstrings and examples here: https://pytorch-lightning.readthedocs.io/en/latest/common/lightning_module.html#optimizer-step

Replies: 1 comment

Comment options

You must be logged in to vote
0 replies
Answer selected by carmocca
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment