search

advimman / lama

🦙 LaMa Image Inpainting, Resolution-robust Large Mask Inpainting with Fourier Convolutions, WACV 2022
https://advimman.github.io/lama-project/
Apache License 2.0
7.9k stars 838 forks source link

lama train #239

Closed yangjq321 closed 10 months ago

yangjq321 commented 1 year ago

May I ask why I don't save the models for each stage when training my dataset in Lama, and the last saved model is a 0 file? What should I do if I want to save the models for each stage and the saved model is a pth file? thanks

image
senya-ashukha commented 10 months ago

you can add saving in training loop!

yangjq321 commented 10 months ago

您可以在训练循环中添加保存！

Thank you very much!