odegeasslbc / FastGAN-pytorch

Official implementation of the paper "Towards Faster and Stabilized GAN Training for High-fidelity Few-shot Image Synthesis" in ICLR 2021
GNU General Public License v3.0
600 stars 100 forks source link

KeyError: 'init.init.0.weight_orig' #39

Open yangyu615 opened 2 years ago

yangyu615 commented 2 years ago

When I run the eval.py, there is something wrong with exit code 1:KeyError: 'init.init.0.weight_orig'

flh155 commented 1 year ago

try it: xxx.load_state_dict(checkpoint['g'])---->xxx.load_state_dict(checkpoint['g'],strict=False)