Update BidirectionalTranslation/options/base_options.py

BidirectionalTranslation/options/base_options.py

@@ -20,7 +20,7 @@ class BaseOptions():
         parser.add_argument('--input_nc', type=int, default=1, help='# of input image channels')     # Modified default
         parser.add_argument('--output_nc', type=int, default=3, help='# of output image channels')   # Modified default
         parser.add_argument('--nz', type=int, default=64, help='#latent vector')                     # Modified default
-        parser.add_argument('--gpu_ids', type=str, default='0', help='gpu ids: e.g. 0  0,1,2, 0,2, -1 for CPU mode')
+        parser.add_argument('--gpu_ids', type=str, default='-1', help='gpu ids: e.g. 0  0,1,2, 0,2, -1 for CPU mode')
         parser.add_argument('--name', type=str, default='color2manga_cycle_ganstft', help='name of the experiment')  # Modified default
         parser.add_argument('--preprocess', type=str, default='none', help='not implemented')         # Modified default
         parser.add_argument('--dataset_mode', type=str, default='aligned', help='aligned,single')
@@ -138,7 +138,7 @@ class BaseOptions():
             id = int(str_id)
             if id >= 0:
                 opt.gpu_ids.append(id)
-        if len(opt.gpu_ids) > 0:
+        if len(opt.gpu_ids) > 0 and torch.cuda.is_available():
             torch.cuda.set_device(opt.gpu_ids[0])
 
         self.opt = opt