search

marialeyvallina / generalized_contrastive_loss

MIT License
87 stars 14 forks source link

Where does `create_model` come from? #14

Closed tmelanson17 closed 1 year ago

tmelanson17 commented 1 year ago

I see a function create_model being called in the [extract_predictions] file, but I'm not sure where it comes from.

I tried using TIMM's create_model, but it seems to load an incompatible state dictionary. Is there another create_model I should be looking at? What version of timm do you use?

marialeyvallina commented 1 year ago

Hi @tmelanson17 Create model comes from src/factory.py We do not use timm in this library. I hope this solves your doubt!