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
8.14k stars 864 forks source link

Question about model size #281

Open dy2728 opened 11 months ago

dy2728 commented 11 months ago

Hi, I have trained lama-fourier model on my own dataset. I found that the saved model size is 676 MB. I thought the model size is too large. May I set any parameters to ignore discriminators or evaluators when saving the model?

jiaxue-ai commented 8 months ago

you can reduce the checkpoint size by converting it from .ckpt to .pt model