Is it possible to add chop_forward

The-Learning-And-Vision-Atelier-LAVA / SOF-VSR

[ACCV 2018 & TIP 2020] Deep Video Super-resolution using HR Optical Flow Estimation

169 stars 43 forks source link

Is it possible to add chop_forward #3

Closed ABDOELSHEMY closed 5 years ago

ABDOELSHEMY commented 5 years ago

Hi @LongguangWang i was wonder if it possible to add chop_forward to this great project to avoid run out of memory and Freezing the pc,i don't know if it even possible with this kind of Video Super-Resolution. thank you

LongguangWang commented 5 years ago

Hi @ABDOELSHEMY I've just updated the code and added a memory-efficient mode when running on GPU. Please try it.

ABDOELSHEMY commented 5 years ago

unfortunately i can not try it because pytorch no longer support my old 650tiboost gpu, i think it is time to buy new gpu to use this amazing project as it Results Much better than (srgan sr) in my tests. thank you very much guys for your time and effort you spent to make this stunning work