lsc-cnn icon indicating copy to clipboard operation
lsc-cnn copied to clipboard

Published 20 hours ago • verified publisher icon val-iisc

How can we test the code on CPU only ?

Open MounirB opened this issue 4 years ago • 1 comments

I tried to replace all the torch instructions that include cuda in the main script, but each time I run the code to test a model, it looks for a GPU. Can you tell me how can I run the code on CPU only to test a model ?

MounirB avatar Oct 02 '19 08:10 MounirB

Could you paste here the error that you'd got? Can you also share the script that you'd used to run?

pvskand avatar Oct 02 '19 13:10 pvskand