only cpu

[xc597]

When running test.py, can I only set the CPU to run? If so, how should I do it?

[layumi]

Hi @xc597
I think it is Okay. You need to remove all .cuda() and specify .cpu().