YOLO-World3

Sleeping

csmithxc commited on Apr 19

Commit

a2c2c3a

•

1 Parent(s): 45f7870

Update configs/finetune_coco/yolo_world_s_t2i_bn_2e-4_100e_4x8gpus_coco_finetune.py

Files changed (1) hide show

configs/finetune_coco/yolo_world_s_t2i_bn_2e-4_100e_4x8gpus_coco_finetune.py CHANGED Viewed

@@ -32,7 +32,7 @@ model = dict(
         image_model={{_base_.model.backbone}},
         text_model=dict(
             type='HuggingCLIPLanguageBackbone',
-            model_name='pretrained_models/clip-vit-base-patch32-projection',
             frozen_modules=['all'])),
     neck=dict(type='YOLOWorldPAFPN',
               guide_channels=text_channels,

         image_model={{_base_.model.backbone}},
         text_model=dict(
             type='HuggingCLIPLanguageBackbone',
+            model_name='openai/clip-vit-base-patch32-projection',
             frozen_modules=['all'])),
     neck=dict(type='YOLOWorldPAFPN',
               guide_channels=text_channels,