init

APDrawingGAN2/options/base_options.py

@@ -178,7 +178,7 @@ class BaseOptions():
             if id >= 0:
                 opt.gpu_ids.append(id)
         if len(opt.gpu_ids) > 0:
-            #torch.cuda.set_device(opt.gpu_ids[0])
+            torch.cuda.set_device(opt.gpu_ids[0])
         
         # set gpu ids
         str_ids = opt.gpu_ids_p.split(',')
@@ -186,7 +186,7 @@ class BaseOptions():
         for str_id in str_ids:
             id = int(str_id)
             if id >= 0:
-                #opt.gpu_ids_p.append(id)
+                opt.gpu_ids_p.append(id)
 
         self.opt = opt
         return self.opt