-
Notifications
You must be signed in to change notification settings - Fork 12
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
Comments
@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. |
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 |
Hi! I wonder how many GPUs and the batch size per GPU you used? @raindrop313 |
hellow, May I ask if you can share your training time? I trained with 4090 GPU for 10 hours and only completed 65k iterations |
Hello, I obtained a PSNR of only 33 when training on the indoor dataset using your model. Can you please explain why?
The text was updated successfully, but these errors were encountered: