towhee-io / examples

Analyze the unstructured data with Towhee, such as reverse image search, reverse video search, audio classification, question and answer systems, molecular search, etc.
Apache License 2.0
447 stars 112 forks source link

Loading Local Models and Downloading Online Models #225

Closed tian-003 closed 1 year ago

tian-003 commented 1 year ago

Hello. When I use ''ops.image_embedding.timm(model_name='resnet50', checkpoint_path=my_checkpoint_path)'' , even if I specify the ''checkpoint_path='' to a specific model path, it still checks if the pre-trained model exists in the [ .cache] directory. If it is not present, the function automatically downloads the pre-trained model. In this case, is it actually using the model I specified in 'checkpoint_path=' ? Thanks for you!

jaelgu commented 1 year ago

Fixed by latest timm operator. Use the code to force re-download the latest operator: ops.image_embedding.timm(xxx).latest()

tian-003 commented 1 year ago

Thank you! The problem has been solved! Good!