jolibrain/joliGEN

Resuming a CUT model fails

beniz opened this issue · 1 comments

beniz commented

I get this error:

initialize network with normal
initialize network with normal
model [CUTSemanticMaskModel] was created
loading the model from /data1/beniz/models/gan_checkpoints/face_masks_cut_2/latest_net_G.pth
loading the model from /data1/beniz/models/gan_checkpoints/face_masks_cut_2/latest_net_F.pth
Traceback (most recent call last):
  File "train.py", line 35, in <module>
    model.setup(opt)               # regular setup: load and print networks; create schedulers
  File "/home/beniz/projects/deepdetect/dev/joliGAN/models/base_model.py", line 137, in setup
    self.load_networks(load_suffix)
  File "/home/beniz/projects/deepdetect/dev/joliGAN/models/base_model.py", line 254, in load_networks
    self.__patch_instance_norm_state_dict(state_dict, net, key.split('.'))
  File "/home/beniz/projects/deepdetect/dev/joliGAN/models/base_model.py", line 230, in __patch_instance_norm_state_dict
    self.__patch_instance_norm_state_dict(state_dict, getattr(module, key), keys, i + 1)
  File "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py", line 948, in __getattr__
    type(self).__name__, name))
AttributeError: 'PatchSampleF' object has no attribute 'mlp_0'

Probably some of the network parameters that are not saved.

it will be fixed with #29 !