Question about model size

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.91k stars 838 forks source link

Question about model size #281

Open dy2728 opened 10 months ago

dy2728 commented 10 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 6 months ago

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