Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Releasing the model on torch.hub? #1

Open
gmberton opened this issue Oct 6, 2024 · 2 comments
Open

Releasing the model on torch.hub? #1

gmberton opened this issue Oct 6, 2024 · 2 comments

Comments

@gmberton
Copy link

gmberton commented Oct 6, 2024

Hey @Ahmedest61, it was great meeting you at ECCV!

Are there any plans to release the trained model on torch.hub? It is quite simple to do and allows people to use your model with two lines of code, allowing more people to use your model and helping to spread your work!
For example I did it for CosPlace, and the trained model can be automatically downloaded from anywhere without cloning the repo just like this

import torch
model = torch.hub.load("gmberton/cosplace", "get_trained_model", backbone="ResNet50", fc_output_dim=2048)

I'd like to add your model to the VPR-methods-evaluation repo and having the model on torch.hub would greatly help

@Ahmedest61
Copy link
Owner

Hey @gmberton,

It was great meeting you at ECCV as well, and thank you for your suggestion!

Adding the trained model to Torch Hub is indeed on my to-do list, and I completely agree that it would make the model more accessible and easier to evaluate. I'll keep you posted once I make more progress on this!

@Ahmedest61
Copy link
Owner

Hey @gmberton, I've added the trained model on Torch Hub.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants