'Pix2PixHDModel' object has no attribute 'module'

[yejg2017]

When I run my own image data,I met this bug,and how can I solved this problem?
type(self).name, name))
AttributeError: 'Pix2PixHDModel' object has no attribute 'module'

[tcwang0509]

Which line in which file does this occur?

[alwynmathew]

I couldn’t find a module in model. Could you point me to the code section where its defined.
AttributeError: 'model' object has no attribute 'module'

[oatmeal3000]

When I use cpu only without GPU

create web directory ./checkpoints/melody/web...
Traceback (most recent call last):
File "train.py", line 48, in
optimizer_G, optimizer_D = model.module.optimizer_G, model.module.optimizer_D
File "/usr/local/lib/python3.7/site-packages/torch/nn/modules/module.py", line 772, in getattr
type(self).name, name))
torch.nn.modules.module.ModuleAttributeError: 'Pix2PixHDModel' object has no attribute 'module'

[joel1895]

I am also facing the error, any solution for this?

[kondela]

Any chance you are trying to run it using PyTorch >= 1.0 and on cpu? Try PyTorch 1.5/1.6 and a GPU

[AyeshaShafique]

i also need solution @joel1895

[alwynmathew]

Set gpu_ids as -1 to run in cpu.

pix2pixHD/models/models.py

Line 17 in 5a2c872

if opt.isTrain and len(opt.gpu_ids) and not opt.fp16:

if gpu_ids is not set to -1, model will be wrapped in DataParallel

pix2pixHD/models/models.py

Line 18 in 5a2c872

model = torch.nn.DataParallel(model, device_ids=opt.gpu_ids)

when its wrapped in DataParallel, everything in model can be only access like model.module.something

Hope this helped.

[kex243]

You can remove all instances of ".module" in train.py if commenting in models.py if opt.isTrain and len(opt.gpu_ids) and not opt.fp16: model = torch.nn.DataParallel(model, device_ids=opt.gpu_ids)