diff options
| author | LambdaWill <574819595@qq.com> | 2017-10-21 17:06:30 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2017-10-21 17:06:30 +0800 |
| commit | 9d0295bbb5d01ce40bd2792f72882128282f0338 (patch) | |
| tree | 6910ec980722ece36e3d6a06b0543baf61ff2e81 | |
| parent | 1e0fbeaf5bc81c0b9e23655dc31d28c77505b7cd (diff) | |
Update networks.py
fix typo
| -rw-r--r-- | models/networks.py | 2 |
1 files changed, 1 insertions, 1 deletions
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 |