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

About training on the winter2summer dataset #12

Closed
ZhenyuLiu-SYSU opened this issue Apr 21, 2022 · 4 comments
Closed

About training on the winter2summer dataset #12

ZhenyuLiu-SYSU opened this issue Apr 21, 2022 · 4 comments

Comments

@ZhenyuLiu-SYSU
Copy link

Hello!
Thank you for doing such a good job.
And I am trying to train it on the winter2summer dataset, the settings are as yours.
I use the command --dataroot ./datasets/winter2summer --name winter2summer_SCL --model sc --learned_attn --augment to train the LSeSim model. I trained it about 50epoch and like this
image
the FID=104.3
But the effect is not as good as shown in your paper
image
Could you tell me what's wrong with me?

Thank you!

@lyndonzheng
Copy link
Owner

Hi @ZhenyuLiu-BJFU, thanks for your interesting. I have not trained the winter2summer on this code. I used the MUNIT(https://github.com/NVlabs/MUNIT) for winter2summer translation.
For the results, it seems the structure is ensured well, while the appearance holds many artifacts that are not judged well by the discriminator. It may be better to add more weight for the discriminator loss, because the MUNIT used multi-scales discriminators with more weights on the GAN loss.

@ZhenyuLiu-SYSU
Copy link
Author

Got it!
I try to train more time(about 90 epoch) and the result is
image
Most of the pictures have blue parts, I think it's a little strange.
Of course, I used FID to test the effect and found that FID=87.8, it seems not bad.
What do you think of this effect?

Thank you!

@lyndonzheng
Copy link
Owner

Hi @ZhenyuLiu-BJFU I have not noticed you training epoch before. Generally, the default setting would be 200 times for fixed unpaired I2I translation, while the learnable one following the setting of CUT that use 400 epochs during the training.

@ZhenyuLiu-SYSU
Copy link
Author

Thanks for your explanation.
I will try it again.

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