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
-
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