Merge pull request NVIDIA#2 from TheAussieStew/master

Fix unexpected keyword argument
Alex-CruzG · Dec 10, 2017 · f2bbfe7 · f2bbfe7
2 parents db4a24d + 5b26d6a
commit f2bbfe7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/models/networks.py b/models/networks.py
@@ -41,7 +41,7 @@ def define_G(input_nc, output_nc, ngf, netG, n_downsample_global=3, n_blocks_glo
     print(netG)
     if len(gpu_ids) > 0:
         assert(torch.cuda.is_available())   
-        netG.cuda(device_id=gpu_ids[0])
+        netG.cuda(gpu_ids[0])
     netG.apply(weights_init)
     return netG