search

sczhou / ProPainter

[ICCV 2023] ProPainter: Improving Propagation and Transformer for Video Inpainting
https://shangchenzhou.com/projects/ProPainter/
Other
5.45k stars 645 forks source link

Training with fp16? #78

Open longyangqi opened 5 months ago

longyangqi commented 5 months ago

Great work! However, training on my own data consumes too much GPU memory. As fp16 is supported in Memory-efficient inference, can we also train the model with fp16?

Thanks!