Remove GPU

asp/models/networks.py

@@ -205,7 +205,7 @@ def init_net(net, init_type='normal', init_gain=0.02, gpu_ids=[], debug=False, i
 
     Return an initialized network.
     """
-    if len(gpu_ids) > 0:
+    if gpu_ids is not None and len(gpu_ids) > 0:
         assert(torch.cuda.is_available())
         net.to(gpu_ids[0])
         # if not amp:

main.py

@@ -21,7 +21,7 @@ def convert_he2ihc(input_he_image_path):
     input_img = Image.open(input_he_image_path).convert('RGB')
 
     opt = SimpleNamespace(
-        gpu_ids=[0],
+        gpu_ids=None,
         isTrain=False,
         checkpoints_dir="../../checkpoints",
         # name="ASP_pretrained/BCI_her2_lambda_linear",