search

tarun005 / FLAVR

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

The training batch size of 8x VFI on GoPro #31

Closed hityzy1122 closed 2 years ago

hityzy1122 commented 2 years ago

Hi, What is the batch size for training FLAVR for 8x VFI, I see in the paper that it is 32 with the frame size of 512x512. But I train on 8 GPUs(1080Ti whose memory is the same as 2080Ti), I got OOM error and only a batch size of 16 is ok. Besides that, neither random frame order reversal nor random horizontal flipping as augmentation strategies can be found in the GoPro.py. I wonder that can I reproduce the results with this code ?

tarun005 commented 2 years ago

Hi, thanks for your interest in our work. The GoPro code can be used for inference. For training you have to modify it accordingly adding augmentation etc.

hityzy1122 commented 2 years ago

Thanks, I will have a try