run the code on cpu #38
Description
I'm getting inference from the model, but I want to run it on the CPU. How can I set the model to not use the GPU?
Because I didn't see any part of the code that sets the flag for GPU or CPU.
{{ message }}
I'm getting inference from the model, but I want to run it on the CPU. How can I set the model to not use the GPU?
Because I didn't see any part of the code that sets the flag for GPU or CPU.