Fail to load parameters with Spectral norm?

(Wei Sun) #1

I tried to train a generator and save the state_dict. But got error KeyError: 'conv1_1.weight' when load the saved state_dict.

When I printed out keys of saved state_dict, only find conv1_1.weight_orig, conv1_1.weight_u and conv1_1.weight_v.

How can I solve this issue? Or how can I save the trained model to make it loadable? Thanks