bacon icon indicating copy to clipboard operation
bacon copied to clipboard

Published 20 hours ago • verified publisher icon computational-imaging

run on devices without cuda enabled PyTorch

Open aadityacs opened this issue 2 years ago • 1 comments

The code by default assumes that it has a cuda enabled version of PyTorch. This is not so in cases such as on a MacBook. Maybe changing the .cuda() call to .to(device) where device is set based on cuda availability might make the code run without any hiccup on such devices.

aadityacs avatar Mar 06 '22 03:03 aadityacs

Thanks a lot

Avocadoooooo avatar Jul 24 '22 08:07 Avocadoooooo