init

APDrawingGAN2/models/base_model.py

@@ -20,8 +20,8 @@ class BaseModel():
         self.gpu_ids = opt.gpu_ids
         self.gpu_ids_p = opt.gpu_ids_p
         self.isTrain = opt.isTrain
-        self.device = torch.device('cuda:{}'.format(self.gpu_ids[0])) if self.gpu_ids else torch.device('cpu')
-        self.device_p = torch.device('cuda:{}'.format(self.gpu_ids_p[0])) if self.gpu_ids else torch.device('cpu')
+        self.device = torch.device('cpu')
+        self.device_p = torch.device('cpu')
         self.save_dir = os.path.join(opt.checkpoints_dir, opt.name)
         self.auxiliary_dir = os.path.join(opt.checkpoints_dir, opt.auxiliary_root)
         if opt.resize_or_crop != 'scale_width':

APDrawingGAN2/options/base_options.py

@@ -171,11 +171,11 @@ class BaseOptions():
             self.print_options(opt)
 
         # set gpu ids
-        opt.gpu_ids = None
+        opt.gpu_ids = False
         
         
         # set gpu ids
-        opt.gpu_ids_p = None
+        opt.gpu_ids_p = False
         
 
         self.opt = opt