DINO-Mix icon indicating copy to clipboard operation
DINO-Mix copied to clipboard

Releasing the model on torch.hub?

Open gmberton opened this issue 4 months ago • 0 comments

Hi @GaoShuang98 thank you for open sourcing your model!

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

gmberton avatar Oct 06 '24 00:10 gmberton