From 5b26d6a8baed25641b047d08119fc51a4435299b Mon Sep 17 00:00:00 2001 From: Conway Ying Date: Sun, 10 Dec 2017 06:03:47 +0000 Subject: [PATCH] Fix unexpected keyword argument --- models/networks.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models/networks.py b/models/networks.py index c583b991..2ec23ae9 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