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.