init

APDrawingGAN2/models/networks.py

@@ -63,10 +63,6 @@ def init_weights(net, init_type='normal', gain=0.02):
 
 
 def init_net(net, init_type='normal', init_gain=0.02, gpu_ids=[]):
-    if len(gpu_ids) > 0:
-        assert(torch.cuda.is_available())
-        net.to(gpu_ids[0])
-        net = torch.nn.DataParallel(net, gpu_ids)
     init_weights(net, init_type, gain=init_gain)
     return net
 

app.py

@@ -167,8 +167,8 @@ def main():
     opt.netG = 'resnet_9blocks'
     opt.which_epoch = 150
     opt.how_many = 1000
-    opt.gpu_ids = '-1'
-    opt.gpu_ids_p = '-1'
+    opt.gpu_ids = ''
+    opt.gpu_ids_p = ''
     opt.imagefolder = 'images-single'
 
     opt.checkpoints_dir = checkpoint_dir