CPU compatible

NNfunctions.py

@@ -266,10 +266,7 @@ def EvaluateModel(net,opt,stack):
     inputimg = inputimg.unsqueeze(0)
 
     with torch.no_grad():
-        if opt.cpu:
-            sr = net(inputimg)
-        else:
-            sr = net(inputimg.cuda())
+        sr = net(inputimg.to(opt.device))
         sr = sr.cpu()
         sr = torch.clamp(sr,min=0,max=1)
         print('min max',inputimg.min(),inputimg.max())