search

tarun005 / FLAVR

Code for FLAVR: A fast and efficient frame interpolation technique.
Apache License 2.0
430 stars 69 forks source link

Training issue #46

Closed weiMytian closed 3 months ago

weiMytian commented 2 years ago

Hi, author, thank you for sharing the code on GitHub. The code performed well in test, but the PSNR value was always maintained at about 17dB during training. What is the reason?

ss00atbupt commented 2 years ago

I‘ve met the same question. I adjusted my lr, epoch 0 seems good ,psnr is 20. But epoch 1 decreased to about 17. And how is you training now? b653e2bb0f26b519b29af14c9570a25 a2150f23193b45435565496755e369d

ss00atbupt commented 2 years ago

Hey, I checked my code. May I ask your batch_size value? I set my lr==0.0002 and batch_size==4, psnr value seems normal now. Maybe you can try change your batch_size bigger. Hope it helps.

tarun005 commented 2 years ago

Can you make sure that the versions of the PyTorch are the same as recommended? I did not face any such issues at my end.