How to save fine tuned PaliGemma model?

google-research / big_vision

Official codebase used to develop Vision Transformer, SigLIP, MLP-Mixer, LiT and more.

Apache License 2.0

2.04k stars 140 forks source link

How to save fine tuned PaliGemma model? #104

Closed SkalskiP closed 2 months ago

SkalskiP commented 2 months ago

Hi 👋🏻

Thanks a lot for PaliGemma fine-tune example notebook. It worked without any issue. However, it lacks the last step - saving a fine-tuned model. (Or I'm just missing something.) I'd appreciate any help. Thank you. 🙏🏻

akolesnikoff commented 2 months ago

Thanks for your question!

I have added an extra colab cell to save the last checkpoint.

SkalskiP commented 2 months ago

Awesome, @akolesnikoff 🔥 I really appreciate your help!

milaomrani commented 1 month ago

OverflowError: serializing a bytes object larger than 4 GiB requires pickle protocol 4 or higher