search

chaofengc / PSFRGAN

PyTorch codes for "Progressive Semantic-Aware Style Transformation for Blind Face Restoration", CVPR2021
Other
370 stars 68 forks source link

epoch num #33

Closed zhangyunming closed 3 years ago

zhangyunming commented 3 years ago

hi，in your train code the dauflt total epoch num is 50, but it cost even 20 days on my two 2080ti, and the model you given is 15 epoches ,does 15 is enough? 50 will have a better reslut than 15?

chaofengc commented 3 years ago

Yes, 15 epochs is enough. Training for more epochs does not bring much visual improvement. However, since I did not conduct full test for models with longer training. I am not sure whether it improves the generalization ability.