Skip to content
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

Evaluation and saving the best checkpoint #209

Open
wants to merge 2 commits into
base: master
Choose a base branch
from

Conversation

YCAyca
Copy link

@YCAyca YCAyca commented Sep 26, 2024

Evaluation using IS, KID, FID metrics and saving the best model for inference time added. Utils/utils.py includes the metric evaluation and saving, therefore it can be added with a couple of lines for any type of gan training. Check gan/gan.py for the example adaptation.

@baicaiPCX
Copy link

baicaiPCX commented Sep 26, 2024 via email

…trics and inference: each of them can be used for different type of datasets (mnist or celeba), use the evaluation and make inference
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants