search

myungsub / CAIN

Source code for AAAI 2020 paper "Channel Attention Is All You Need for Video Frame Interpolation"
MIT License
323 stars 43 forks source link

Training issue on Vimeo Dataset #11

Closed daisily727 closed 2 years ago

daisily727 commented 3 years ago

Hi, thanks for sharing the code. I tried to train on Vimeo dataset from scratch just use your script but seems it stucks at PSNR around 15 and SSIM around 0.5. Is there any special recipe that I am mising in the training process?

myungsub commented 3 years ago

Hi, thanks for your interest in our work. There shouldn't be any special recipes.. could you tell us about your problem more in detail? Does our pretrained model work fine in your running environment?

HubertSotnowski commented 3 years ago

Hi, Training cain on new PyTorch don't work correctly. Try PyTorch 1.5.

daisily727 commented 3 years ago

Hi, Training cain on new PyTorch don't work correctly. Try PyTorch 1.5.

Thanks for the reply. I have met the same. I got it work correctly now by changing from 1.7 to 1.4