How do I push a whole dataloader through a LightningModule? #6489
Unanswered
wittenator
asked this question in
Lightning Trainer API: Trainer, LightningModule, LightningDataModule
Replies: 0 comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Uh oh!
There was an error while loading. Please reload this page.
-
I migrated my model without much problems to Pytorch Lightning. Abstracting the train/eval/test loops makes sense and shortens my code significantly. Now I am wondering though how I can push a whole dataloader through my model without writing the loops myself and transferring the batches to the corresponding devices. Writing that manually doesn't seem to be in the vein of Pytorch Lightning. Is there a stylistically correct way to do that?
Beta Was this translation helpful? Give feedback.
All reactions