code implementation

[sujanme25]

Thank you for the code.

How can i run in cpu? the code only support CUDA?

Regards

[JavierAntoran]

Hi @sujanme25,

The training scripts should autodetect whether you are on CPU or GPU. The code should work on both out of the box. If you encounter that this is not the case, could you send the command that you are running so I can reproduce the issue?

Best,
Javier