TypeError: cuda() got an unexpected keyword argument 'device_id'

[zxt-triumph]

I encountered a problem while training.
Traceback (most recent call last):
File "train.py", line 13, in
model = create_model(opt)
File "/home/zxt/GAN Network/PS-GAN/models/models.py", line 19, in create_model
model.initialize(opt)
File "/home/zxt/GAN Network/PS-GAN/models/pix2pix_model.py", line 36, in initialize
opt.which_model_netG, opt.norm, not opt.no_dropout, self.gpu_ids)
File "/home/zxt/GAN Network/PS-GAN/models/networks.py", line 51, in define_G
netG.cuda(device_id=gpu_ids[0])
TypeError: cuda() got an unexpected keyword argument 'device_id'
How to solve this problem？ Looking forward to your reply.

[yifanjiang19]

It’s due to the pytorch version. Just replace ‘device_id’ with ‘divice’.

[zxt-triumph]

Thank you for your reply! But I meet new problem.
Traceback (most recent call last):
File "train.py", line 13, in
model = create_model(opt)
File "/home/image/zxt/PS-GAN/models/models.py", line 19, in create_model
model.initialize(opt)
File "/home/image/zxt/PS-GAN/models/pix2pix_model.py", line 43, in initialize
self.netD_person = networks.define_person_D(opt.input_nc, opt.ndf, opt, use_sigmoid, self.gpu_ids)
File "/home/image/zxt/PS-GAN/models/networks.py", line 85, in define_person_D
netD = PersonDiscriminator(input_nc, ndf, use_sigmoid, gpu_ids=gpu_ids)
TypeError: init() got multiple values for keyword argument 'gpu_ids'

[yifanjiang19]

Could you please try to replace it with PersonDiscriminator(input_nc, ndf, gpu_ids=gpu_ids) ?

[zxt-triumph]

Thank you for your help!