Bayesian-Neural-Networks icon indicating copy to clipboard operation
Bayesian-Neural-Networks copied to clipboard

Published 20 hours ago • verified publisher icon JavierAntoran

code implementation

Open sujanme25 opened this issue 3 years ago • 1 comments

Thank you for the code.

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

Regards

sujanme25 avatar Jan 19 '22 13:01 sujanme25

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

JavierAntoran avatar Aug 16 '22 17:08 JavierAntoran