feat(config): 添加图像检索任务并更新模型配置

- 在默认配置中添加了图像检索任务(retrieval)
- 更新了模型配置文件，启用了CUHK-PEDES数据集
- 注释掉了GCC和RedCAP数据集
- 移除了模型打印语句
- 修复了分布式启动脚本的参数传递问题

configs/default.yml

@@ -94,6 +94,7 @@ evaluate:
   task:
     - cls
     - seg
+    - retrieval
   cls:
     save_best: true
     template: subset
@@ -102,6 +103,11 @@ evaluate:
     cfg: segmentation/configs/_base_/datasets/pascal_voc12.py
     template: simple
     opts: []
+  retrieval:
+    save_best: true
+    template: simple
+    opts: []
+    
 
 checkpoint:
   auto_resume: true

configs/group_vit_gcc_redcap_cuhkpedes_30e.yml

@@ -24,6 +24,7 @@ model:
 data:
   dataset:
     train:
-      - gcc3m
-      - gcc12m
-      - redcap12m
+      # - gcc3m
+      # - gcc12m
+      # - redcap12m
+      - cuhkpedes_train

models/builder.py

@@ -18,6 +18,6 @@ def build_model(config):
 
     model = MODELS.build(OmegaConf.to_container(config, resolve=True))
     
-    print(model)
+    # print(model)
 
     return model

tools/dist_launch.sh

@@ -19,4 +19,4 @@ PORT=${PORT:-29500}
 
 PYTHONPATH="$(dirname $0)/..":$PYTHONPATH \
 python -m torch.distributed.launch --nproc_per_node=$GPUS --master_port=$PORT \
-    $SCRIPT --cfg $CONFIG \ --resume $RESUME ${@:5}
+    $SCRIPT --cfg $CONFIG --resume $RESUME ${@:5}