Closed narminGhaffari closed 2 months ago
Hi, I think there are two issues, the --pretrained openai
flag makes you load models that were pretrained by openai while coca is not one of those, at the same time I think currently coca training is broken, you can try to have a look at this PR for a fix of #710
I am using the following script to train a fine-tune the coco model:
However, I receive:
Model coca_ViT-B-32 not found; available models = ['RN50', 'RN50-quickgelu', 'RN101', 'RN101-quickgelu', 'RN50x4', 'RN50x16', 'RN50x64', 'ViT-B-32', 'ViT-B-32-quickgelu', 'ViT-B-16', 'ViT-L-14', 'ViT-L-14-336']
I am using open_clip version 2.24.0. Can someone please help me to solve this issue?