DDP not moving everything to GPU #17554
Unanswered
surya-narayanan
asked this question in
DDP / multi-GPU / multi-node
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 have code that has the following structure
it seems
self.encoder.model
andself.decoder.model
are not moved to the GPU, and stay on cpu. Is this expected?Beta Was this translation helpful? Give feedback.
All reactions