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

Discriminator's impact on recons loss. #350

Open
NITHISHM2410 opened this issue Feb 18, 2024 · 0 comments
Open

Discriminator's impact on recons loss. #350

NITHISHM2410 opened this issue Feb 18, 2024 · 0 comments

Comments

@NITHISHM2410
Copy link

NITHISHM2410 commented Feb 18, 2024

I used VQ f4 config ( 256 -> 64 ) in this repo to train VQGAN(taming transformers) model on COCO dataset. My recons loss steadily decreases until the discriminator kicks in. I trained my model for 50,000 iterations (54 epochs ) without disc, the reconstructions were good but bit blurry. So I decided to bring in Discriminator around 51000 iterations, since then the recons loss started to increase losing image quality.
Before the disc entry, the d_weight value was around 25-30 and recons loss (mae + lpips) was around 0.15 & I set the disc_weight to 0.75 , after disc entry the recons loss increased to 0.4 and oscillates around.
Anyone know a solution ?

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

1 participant