Mixed Precision Training (Apex)

xinntao / EDVR

Winning Solution in NTIRE19 Challenges on Video Restoration and Enhancement (CVPR19 Workshops) - Video Restoration with Enhanced Deformable Convolutional Networks. EDVR has been merged into BasicSR and this repo is a mirror of BasicSR.

https://github.com/xinntao/BasicSR

1.48k stars 320 forks source link

Mixed Precision Training (Apex) #98

Closed TouqeerAhmad closed 4 years ago

TouqeerAhmad commented 4 years ago

Hi, have anybody looked into training EDVR using Apex API -- I have been trying but getting issues due to DCNv2. Wanted to know if anyone else has tried or have some wisdom to share.

xinntao commented 4 years ago

Thanks @TouqeerAhmad We now use the mmdection's dcn implementation. mmdetection supports fp16 training. Maybe there is a solution in mmdection.

ShihuaHuang95 commented 4 years ago

@TouqeerAhmad Have you found any solution to this issue?