Ignore submodule while loading onto the GPU. #18211
Unanswered
konrad-karanowski
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.
Uh oh!
There was an error while loading. Please reload this page.
-
Hello. I have a following code:
`class Lightning(LightningModule):
I want parameters of
self.net2
to be ignored while loading model onto the GPU during training. How can I achieve this?Beta Was this translation helpful? Give feedback.
All reactions