Completely fix unexpected keyword argument

Alex-CruzG · Dec 11, 2017 · 25da17a · 25da17a
1 parent f2bbfe7
commit 25da17a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/models/networks.py b/models/networks.py
@@ -51,7 +51,7 @@ def define_D(input_nc, ndf, n_layers_D, norm='instance', use_sigmoid=False, num_
     print(netD)
     if len(gpu_ids) > 0:
         assert(torch.cuda.is_available())
-        netD.cuda(device_id=gpu_ids[0])
+        netD.cuda(gpu_ids[0])
     netD.apply(weights_init)
     return netD