deepglint / unicom

MLCD & UNICOM : Large-Scale Visual Representation Model
https://huggingface.co/DeepGlint-AI/mlcd-vit-large-patch14-336
322 stars 18 forks source link

Training mode #15

Open vioanl opened 1 year ago

vioanl commented 1 year ago

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

kongzijian commented 1 year ago

train command in scriptts dir