From 6f00fc795473ef2fd3f398a05254a078cb6017af Mon Sep 17 00:00:00 2001 From: jingyaogong Date: Thu, 3 Apr 2025 17:55:43 +0800 Subject: [PATCH] update dataset path --- train_pretrain_vlm.py | 2 +- train_sft_vlm.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/train_pretrain_vlm.py b/train_pretrain_vlm.py index 0bc7766..d185f14 100644 --- a/train_pretrain_vlm.py +++ b/train_pretrain_vlm.py @@ -147,7 +147,7 @@ if __name__ == "__main__": parser.add_argument("--use_wandb", default=False, action="store_true") parser.add_argument("--wandb_project", type=str, default="MiniMind-V") parser.add_argument("--num_workers", type=int, default=8) - parser.add_argument("--data_path", type=str, default="./dataset/pretrain_vlm_data.jsonl") + parser.add_argument("--data_path", type=str, default="./dataset/pretrain_data.jsonl") parser.add_argument("--images_path", type=str, default="./dataset/pretrain_images") parser.add_argument("--ddp", action="store_true") parser.add_argument("--accumulation_steps", type=int, default=1) diff --git a/train_sft_vlm.py b/train_sft_vlm.py index ea33c1a..f847300 100644 --- a/train_sft_vlm.py +++ b/train_sft_vlm.py @@ -137,7 +137,7 @@ if __name__ == "__main__": parser.add_argument("--use_wandb", default=False, action="store_true") parser.add_argument("--wandb_project", type=str, default="MiniMind-V") parser.add_argument("--num_workers", type=int, default=8) - parser.add_argument("--data_path", type=str, default="./dataset/sft_vlm_data.jsonl") + parser.add_argument("--data_path", type=str, default="./dataset/sft_data.jsonl") parser.add_argument("--images_path", type=str, default="./dataset/sft_images") parser.add_argument("--ddp", action="store_true") parser.add_argument("--accumulation_steps", type=int, default=1)