hylee commited on
Commit
fc25463
·
1 Parent(s): 928b52a
APDrawingGAN2/options/base_options.py CHANGED
@@ -23,8 +23,8 @@ class BaseOptions():
23
  parser.add_argument('--netG', type=str, default='unet_256', help='selects model to use for netG')
24
  parser.add_argument('--nnG', type=int, default=9, help='specify nblock for resnet_nblocks, ndown for unet for unet_ndown')
25
  parser.add_argument('--n_layers_D', type=int, default=3, help='only used if netD==n_layers')
26
- parser.add_argument('--gpu_ids', type=str, default='0', help='gpu ids: e.g. 0 0,1,2, 0,2. use -1 for CPU')
27
- parser.add_argument('--gpu_ids_p', type=str, default='0', help='gpu ids for pretrained auxiliary models: e.g. 0 0,1,2, 0,2. use -1 for CPU')
28
  parser.add_argument('--name', type=str, default='experiment_name', help='name of the experiment. It decides where to store samples and models')
29
  parser.add_argument('--dataset_mode', type=str, default='aligned', help='chooses how datasets are loaded. [unaligned | aligned | single]')
30
  parser.add_argument('--model', type=str, default='apdrawing',
 
23
  parser.add_argument('--netG', type=str, default='unet_256', help='selects model to use for netG')
24
  parser.add_argument('--nnG', type=int, default=9, help='specify nblock for resnet_nblocks, ndown for unet for unet_ndown')
25
  parser.add_argument('--n_layers_D', type=int, default=3, help='only used if netD==n_layers')
26
+ parser.add_argument('--gpu_ids', type=str, default='', help='gpu ids: e.g. 0 0,1,2, 0,2. use -1 for CPU')
27
+ parser.add_argument('--gpu_ids_p', type=str, default='', help='gpu ids for pretrained auxiliary models: e.g. 0 0,1,2, 0,2. use -1 for CPU')
28
  parser.add_argument('--name', type=str, default='experiment_name', help='name of the experiment. It decides where to store samples and models')
29
  parser.add_argument('--dataset_mode', type=str, default='aligned', help='chooses how datasets are loaded. [unaligned | aligned | single]')
30
  parser.add_argument('--model', type=str, default='apdrawing',