From 9d0295bbb5d01ce40bd2792f72882128282f0338 Mon Sep 17 00:00:00 2001
From: LambdaWill <574819595@qq.com>
Date: Sat, 21 Oct 2017 17:06:30 +0800
Subject: Update networks.py

fix typo
---
 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 dca5489..949659d 100644
--- a/models/networks.py
+++ b/models/networks.py
@@ -139,7 +139,7 @@ def define_D(input_nc, ndf, which_model_netD,
         raise NotImplementedError('Discriminator model name [%s] is not recognized' %
                                   which_model_netD)
     if use_gpu:
-        netD.cuda(device_id=gpu_ids[0]) # or netG.cuda(device=gpu_ids[0]) for latest version.
+        netD.cuda(device_id=gpu_ids[0]) # or netD.cuda(device=gpu_ids[0]) for latest version.
     init_weights(netD, init_type=init_type)
     return netD
 
-- 
cgit v1.2.3-70-g09d2