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
Uh oh!
There was an error while loading. Please reload this page.
-
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