From 25da17a66bf0915010c74f40697814502f973579 Mon Sep 17 00:00:00 2001
From: Ming <a@ufoym.com>
Date: Mon, 11 Dec 2017 14:11:14 +0800
Subject: Completely fix unexpected keyword argument

---
 models/networks.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'models')

diff --git a/models/networks.py b/models/networks.py
index 2ec23ae..bf9f14b 100755
--- 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
 
-- 
cgit v1.2.3-70-g09d2