What would be moved to GPU by lightning? #15074
Unanswered
marcmk6
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.
-
It's nice that lightning automatically transfer model to GPU, as the starter instruction said
There are no .cuda() or .to(device) calls required. Lightning does these for you., I'm still wondering the implementation detail here, what's the criterion for deciding what to transfer? Having.to()?Beta Was this translation helpful? Give feedback.
All reactions