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

Could not reproduce the results on CIFAR-10 #3

Open
zzw-zwzhang opened this issue May 13, 2022 · 8 comments
Open

Could not reproduce the results on CIFAR-10 #3

zzw-zwzhang opened this issue May 13, 2022 · 8 comments

Comments

@zzw-zwzhang
Copy link

I used your provided model-790000.ckpt on DDIM codebase to generate cifar10 images. But I got the FID is 4.818.

Is there any difference?

@jS5t3r
Copy link

jS5t3r commented May 21, 2023

how many samples did you generate? FID depends on the number of samples.

@zzw-zwzhang
Copy link
Author

how many samples did you generate? FID depends on the number of samples.

Hi, 50000 samples are generated for evaluation.

@jS5t3r
Copy link

jS5t3r commented May 21, 2023

well, that ought to be fine.... I know on that on the paper there was a FID something around 3...
DDIM is not as good as ddpm

@jS5t3r
Copy link

jS5t3r commented May 21, 2023

Did you use the "ema_cifar10"?

@zzw-zwzhang
Copy link
Author

Did you use the "ema_cifar10"?

It's been a long time and I forgot which model I used. But, if the diffusion model is not trained with ema, its FID should be 12.

After that, I trained a DDPM model from scratch, and the fid is close to that in the paper.

@jS5t3r
Copy link

jS5t3r commented May 21, 2023

Would you recommend another repository (maybe with pretrained weights on cifar10 ema)?

@zzw-zwzhang
Copy link
Author

Would you recommend another repository (maybe with pretrained weights on cifar10 ema)?

I forget which codebase provides reliable pre-trained DDPM models. You can try PNDM first. If you cannot reproduce the results, you can train a DDPM model from scratch.

https://github.com/luping-liu/PNDM
https://github.com/zzw-zwzhang/pytorch-ddpim

@jS5t3r
Copy link

jS5t3r commented May 21, 2023

Thanks. Here comes an update. Seen on huggingface.

https://huggingface.co/google/ddpm-cifar10-32/discussions/10#646a1aec96cfe72aef787baf

There is also claimed to use ddim with 100 steps and fid of 4.5.

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

No branches or pull requests

2 participants