lama
lama copied to clipboard
advimman
Cannot fine tune CelebaHq model checkpoint
Hello, I am unable to load celebaHq model.ckpt for fine tuning purpose. It show the below warning even after using "+trainer.kwargs.resume_from_checkpoint=LaMa_models/lama-celeba-hq/lama-fourier/models/best.ckpt" flag for using pretrained checkpoints.
UserWarning: No checkpoint file exists at resume_from_checkpoint. Start from scratch
Kindly help me resolve this issue.
Best regards, Aman
Got the solution for the above problem. But now it is showing another error:
@windj007 Hello, I am currently working on a project where i need to fine tune my model, and i believe that using the LAMA inpainting CelebaHQ model weights files as a starting point could be greatly benefit my research. I kindly request if you would be willing to share the complete CelebaHQ model weight files (both Gen&Disc) with me.
Looking forward for your help and support.
best regards, Aman Gupta
Got the solution for the above problem. But now it is showing another error:
Hi! How did you solve this problem? I have same error message now
@HappyYuji
Hello, I have addressed the "UserWarning: No checkpoint file exists at resume_from_checkpoint. Start from scratch" error by using the absolute path for the weight files.
Regarding the error depicted in the image, I discovered that the pretrained LAMA model does not include discriminator weight files. Instead, it only contains generator weights in the .ckpt file. This means we cannot fine-tune the pretrained model on our dataset. The "best.ckpt" model weights are solely intended for inference purposes.
Furthermore, I believe the authors have only provided a complete checkpoint (both discriminator and generator weights) for the Big LAMA variant.
@aman-autophoto-ai Thank you! Then I also hope that the authors provide a complete checkpoint. Have a nice day.