diff options
| author | Ting-Chun Wang <tcwang0509@berkeley.edu> | 2017-12-11 12:48:19 -0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2017-12-11 12:48:19 -0800 |
| commit | 8133fbe49a3b9d3b813c3c2fd8b761708ed02489 (patch) | |
| tree | ecc53b647b841c9e6b79bf841313e4dcfb7dcf96 | |
| parent | f2bbfe70265263ae43a0a7c1766e98b2f9ef7545 (diff) | |
| parent | 25da17a66bf0915010c74f40697814502f973579 (diff) | |
Merge pull request #3 from ufoym/master
Completely fix unexpected keyword argument
| -rwxr-xr-x | models/networks.py | 2 |
1 files changed, 1 insertions, 1 deletions
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 |