diff --git a/models/networks.py b/models/networks.py
index c583b991..2ec23ae9 100755
--- 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