Load fine tuned model with customised data set in demo

arampacha / CLIP-rsicd

Apache License 2.0

211 stars 31 forks source link

Load fine tuned model with customised data set in demo #40

Closed MichSc33 closed 2 years ago

MichSc33 commented 2 years ago

Hi,

i was able to fine tune a CLIP model on a custom data set with the provided training script. The outcome was a "config.json" as well as a "flax_model.msgpack" file. Is there a way to import my customised model in the provided demo?