Error loading pre-trained checkpoints

[thanh-isu]

Hi,

Thank you for your great work. I downloaded the checkpoints on Google drive and tried to do sampling, but I got this error:

RuntimeError: Error(s) in loading state_dict for DataParallel:
	Missing key(s) in state_dict: "module.normalizer.alpha", "module.normalizer.gamma", "module.normalizer.beta", "module.res1.0.normalize2.alpha", "module.res1.0.normalize2.gamma"
....

I was wondering how you and others load the model wrapped with DataParallel? I can train the model from scratch and fix the saving but wanted to check if there is better workaround.

[yang-song]

I didn't encounter this problem before. Maybe try to downgrade PyTorch to a lower version such as 1.2?