deepglint / unicom

[ICLR 2023] Unicom: Universal and Compact Representation Learning for Image Retrieval

Home Page:https://arxiv.org/pdf/2304.05884.pdf

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Training mode

vioanl opened this issue · comments

commented

I understand that there is an evaluation mode for evaluating the image retrieval using: torchrun` retrieval.py --eval --dataset cub --model_name ViT-B/32
Is there a mode for training the models or should the scripts be executed manually?
Thanks in advance

train command in scriptts dir