Force State to be on CPU #2387
ChristophReich1996
started this conversation in
General
Force State to be on CPU
#2387
Replies: 0 comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
Hi, I'm trying to track a large state in my custom matric. For this reason, I would like to somehow force the state to always be on the CPU. Is there a clean way to do so? I'm using PL for training. Thanks for the help!
Beta Was this translation helpful? Give feedback.
All reactions