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

How to train this model/ #57

Open Junye-Chen opened 10 months ago

Junye-Chen commented 10 months ago

Thank you for your outstanding work. I tried to train the model with the default hyperparameters except batchsize=3 on 3*rtx4090, but I found that the loss values corrupted to NaN at an early stage (about 4000 iters). I would like to have some advise on training skills. Thanks!