Call or Forward? #5811
Answered
by
williamFalcon
keunwoochoi
asked this question in
Lightning Trainer API: Trainer, LightningModule, LightningDataModule
-
Hi, thanks for the nice library. In the |
Beta Was this translation helpful? Give feedback.
Answered by
williamFalcon
Dec 17, 2019
Replies: 1 comment
-
forward should implement what you want to use when calling model(x). you may need to call that in training step (usually do), which means you have to do self.forward(...) because you are in the model when you make that call. |
Beta Was this translation helpful? Give feedback.
0 replies
Answer selected by
carmocca
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
forward should implement what you want to use when calling model(x).
you may need to call that in training step (usually do), which means you have to do self.forward(...) because you are in the model when you make that call.