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

The training results are not satisfactory #5

Open
yydslhk opened this issue Nov 6, 2023 · 4 comments
Open

The training results are not satisfactory #5

yydslhk opened this issue Nov 6, 2023 · 4 comments

Comments

@yydslhk
Copy link

yydslhk commented Nov 6, 2023

Hello, I obtained a PSNR of only 33 when training on the indoor dataset using your model. Can you please explain why?

@raindrop313
Copy link
Collaborator

raindrop313 commented Nov 6, 2023

@yydslhk The psnr result of 33 has obviously not converged yet and you need to try training longer. I usually train 400K iteration with 128 x 128 patch size and 16 batch size, then 800k iteration with 256 x 256 patch size and 8 batch size.

@raindrop313
Copy link
Collaborator

raindrop313 commented Nov 6, 2023

To save time, you can continue to train based on the existing weights, usually I can reach 36+ psnr at tha stage of training with 128*128 patches @yydslhk

@cherrysherryplus
Copy link

cherrysherryplus commented Dec 15, 2023

Hello, I obtained a PSNR of only 33 when training on the indoor dataset using your model. Can you please explain why?

Hi! I wonder how many GPUs and the batch size per GPU you used? @raindrop313

@jg0014
Copy link

jg0014 commented Sep 1, 2024

@yydslhk The psnr result of 33 has obviously not converged yet and you need to try training longer. I usually train 400K iteration with 128 x 128 patch size and 16 batch size, then 800k iteration with 256 x 256 patch size and 8 batch size.

hellow, May I ask if you can share your training time? I trained with 4090 GPU for 10 hours and only completed 65k iterations

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

4 participants