search

FVL2020 / ICCV-2023-MB-TaylorFormer

84 stars 9 forks source link

The training results are not satisfactory #5

Open yydslhk opened 10 months ago

yydslhk commented 10 months ago

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

raindrop313 commented 10 months ago

@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 commented 10 months ago

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 commented 8 months ago

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 commented 5 days ago

@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