-
Notifications
You must be signed in to change notification settings - Fork 1.5k
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
Loss not decreasing #369
Comments
Hey, |
I am training UNet part of latent diffusion with a conditional encoder. I have added one extra module of image encoder for reference. It's been a week I ran the training on almost 3000 images and it has completed the 250 epochs. but the loss is not decreasing, it is still arround 1.0. I am using the 80Gb A100 for training.
Please let me know if I am missing something. or still need to wait for improvements.
I am using the image of size 64643 so not using VAE. I am passing direct image to UNET.
here is loss graph
The text was updated successfully, but these errors were encountered: