The call of training_step and validation_step .etc. #10070
Answered
by
rohitgr7
MrChenFeng
asked this question in
Lightning Trainer API: Trainer, LightningModule, LightningDataModule
-
Anyone help me where did these funcs been called in the core parts? I expect it to be called in the loop instance of trainer however not. Quite confused about this. |
Beta Was this translation helpful? Give feedback.
Answered by
rohitgr7
Oct 21, 2021
Replies: 1 comment
-
depends upon the configuration but in a general case here: https://github.com/PyTorchLightning/pytorch-lightning/blob/454e93bacecb4f1d6ebc9ebd3bc11b26723ddbec/pytorch_lightning/plugins/training_type/training_type_plugin.py#L217-L230 call sequence is |
Beta Was this translation helpful? Give feedback.
0 replies
Answer selected by
tchaton
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
depends upon the configuration but in a general case here: https://github.com/PyTorchLightning/pytorch-lightning/blob/454e93bacecb4f1d6ebc9ebd3bc11b26723ddbec/pytorch_lightning/plugins/training_type/training_type_plugin.py#L217-L230
call sequence is
loop -> accelerators -> training_type_plugin -> actual call