-
Notifications
You must be signed in to change notification settings - Fork 37
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
you should change inception model to evaluation mode before calculating FID score #3
Comments
Hi, thank you for noticing that! I completely forgot about this issue. By the way, while correcting the mode may degrade the absolute FID scores of both fine-tuning and FreezeD, I guess the relative order would remain. Can you report your values if my guess is wrong? |
Yes, relative order remains the same. I'm sorry to tell you that I lost my values with your experiment setting. I will report after I experiment with your setting again. And, thank you for your nice research! |
Happy to hear that the relative orders are the same! |
Hi, do you mean FID score of 49.3 with eval() and 98.1 with train() mode? |
You should change inception model to evaluation mode before calculating FID score.
Inception model contains batch normalization, whose training and evaluation behaviors are different.
For example, you should add "inception.eval()" below line 486 of stylegan/finetune.py
With this correction, I got significantly different FID score comparing to reports from your paper.
The text was updated successfully, but these errors were encountered: