diff options
| author | Jun-Yan Zhu <junyanz@users.noreply.github.com> | 2017-12-09 22:41:18 -0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2017-12-09 22:41:18 -0800 |
| commit | f2bbfe70265263ae43a0a7c1766e98b2f9ef7545 (patch) | |
| tree | 42f3f4b52529493515b5272c18603bb22e1b8552 /models | |
| parent | db4a24df4a8482089bcb88d0e79c9b484307fc75 (diff) | |
| parent | 5b26d6a8baed25641b047d08119fc51a4435299b (diff) | |
Merge pull request #2 from TheAussieStew/master
Fix unexpected keyword argument
Diffstat (limited to 'models')
| -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 c583b99..2ec23ae 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 |