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

Loss does not go down and explodes eventually! #2

Open
slymnyldrm opened this issue Aug 6, 2023 · 1 comment
Open

Loss does not go down and explodes eventually! #2

slymnyldrm opened this issue Aug 6, 2023 · 1 comment

Comments

@slymnyldrm
Copy link

Hello, I am trying to train your model on a single A6000 from scratch with DF2K dataset. First, I have tried your original config and then I have tried batch_size=18 and LR_size=64. Nevertheless, I couldn't train the model. The loss remains at 1 and eventually explodes. Output images is just random rgb noise. Would you be able to help me with this problem?

Thank you

@IceClear
Copy link
Owner

There should be some mistakes in your config. The most possible reason is that you did not load the Autoencoder model correctly.

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