GPU training, CPU using afterwards #26200
Unanswered
jecampagne
asked this question in
Q&A
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.
-
Hello,
If I'm right in Pytorch there is a way to train a model on GPU and save it to be used afterwards on CPU.
Does this mechanism exists in JAX (Flax) too?
Thanks
Beta Was this translation helpful? Give feedback.
All reactions