Update TRAIN_AND_VALIDATE_V1.5.md

PKU-YuanGroup · Jul 21, 2024 · 910a9f6 · 910a9f6
1 parent 24bdd7e
commit 910a9f6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/TRAIN_AND_VALIDATE_V1.5.md b/TRAIN_AND_VALIDATE_V1.5.md
@@ -34,8 +34,8 @@ ChatUniVi/train/train_mem.py \
 --deepspeed scripts/zero3.json \
 --model_name_or_path ${LLM model path} \
 --version v1 \
---model_use PRETUNEv1.5 \
---dataset_use Pretrain \
+--model_use PRETUNE \
+--dataset_use Pretrainv1.5 \
 --vision_tower openai/clip-vit-large-patch14-336 \
 --tune_mm_mlp_adapter True \
 --mm_vision_select_layer -2 \