search

jhgan00 / image-retrieval-transformers

(Unofficial) PyTorch implementation of Training Vision Transformers for Image Retrieval(El-Nouby, Alaaeldin, et al. 2021).
44 stars 6 forks source link

Trained checkpoints #1

Closed somepago closed 1 year ago

somepago commented 1 year ago

Hi,

Thank you so much for implementing the code. I am wondering if you can share the checkpoints.

Thanks, Gowthami

jhgan00 commented 1 year ago

Hi,

Currently, I don't have checkpoints for trained models. So I trained IRTR model with CUB_200_2011 dataset again and uploaded the checkpoint and the training arguments here, and colab example here. However, other datasets are relatively large and it takes several hours to train the model in my environment.

I'm sorry that I can't provide you with other checkpoints. If it is okay to take a long time, it can also provide checkpoints for other datasets within a few days.

Thanks !

somepago commented 1 year ago

Thank you so much. This is good enough. 💯