How to use the pretrained models?

elvisyjlin / AttGAN-PyTorch

AttGAN PyTorch Arbitrary Facial Attribute Editing: Only Change What You Want

MIT License

248 stars 61 forks source link

How to use the pretrained models? #1

Closed 2401ch closed 5 years ago

2401ch commented 5 years ago

I download the pretrained models,but there is no setting.txt. I use the txt thar is created by train.py. However, i can not test by the pretrained models, "Error(s) in loading state_dict for Generator: Unexpected key(s) in state_dict......" How the setting the network about pretrained???

elvisyjlin commented 5 years ago

Hi @chenghan1995 , I'm sorry that I didn't put the setting.txt in both pertained model files. The new zip files have been uploaded. Please download them again. Let me know if you have any other questions.

2401ch commented 5 years ago

Thank you for your valuable work.