'Pix2PixHDModel' object has no attribute 'module' about pix2pixhd HOT 8 OPEN

Set gpu_ids as -1 to run in cpu.

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

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

Hope this helped.

from pix2pixhd.

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)

from pix2pixhd.

Which line in which file does this occur?

from pix2pixhd.

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'

from pix2pixhd.

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'

from pix2pixhd.

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

from pix2pixhd.

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

from pix2pixhd.

i also need solution @joel1895

from pix2pixhd.