Update Scripts/model.py

Scripts/model.py

@@ -33,6 +33,6 @@ def create_cam(model):
     target_layers = [model.net._blocks[-1]]
     targets = [ClassifierOutputTarget(1)]
     cam_algorithm = GradCAMElementWise
-    use_cuda = torch.cuda.is_available() and model.device.type == 'cuda'
+    use_cuda = torch.cuda.is_available() and next(model.parameters()).is_cuda
     cam = cam_algorithm(model=model, target_layers=target_layers, use_cuda=use_cuda)
     return cam