PD-Denoising-pytorch icon indicating copy to clipboard operation
PD-Denoising-pytorch copied to clipboard

Published 20 hours ago • verified publisher icon yzhouas

for cpu

Open SawantMyelin opened this issue 2 years ago • 3 comments

can you please let me know how to test the model on cpu because it asks for a gpu everytime i run the script

SawantMyelin avatar Sep 01 '22 06:09 SawantMyelin

Thanks for your question. This is a very old codebase and the pyTorch is also old. I am not sure whether it will work, but a simple solution would be removing all the .cuda() in the code to start?

yzhouas avatar Sep 01 '22 07:09 yzhouas

Thanks for the response, let me check and i will let you know.

SawantMyelin avatar Sep 01 '22 08:09 SawantMyelin

I got it to work by removing data parallelism line and doing some other modifications in the main.py and removing all the .cuda() lines in other code. The results were amazing. Thanks

SawantMyelin avatar Sep 02 '22 05:09 SawantMyelin