github
/
minimind-v
mirror of https://github.com/jingyaogong/minimind-v.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

update dataset path

pull/45/merge
jingyaogong 2025-04-03 17:55:43 +08:00
parent 9f1d67246b
commit 6f00fc7954
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train_pretrain_vlm.py
View File

@ -147,7 +147,7 @@ if __name__ == "__main__":
parser.add_argument("--use_wandb", default=False, action="store_true") parser.add_argument("--use_wandb", default=False, action="store_true")
parser.add_argument("--wandb_project", type=str, default="MiniMind-V") parser.add_argument("--wandb_project", type=str, default="MiniMind-V")
parser.add_argument("--num_workers", type=int, default=8) 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("--images_path", type=str, default="./dataset/pretrain_images")
parser.add_argument("--ddp", action="store_true") parser.add_argument("--ddp", action="store_true")
parser.add_argument("--accumulation_steps", type=int, default=1) parser.add_argument("--accumulation_steps", type=int, default=1)

2
train_sft_vlm.py
View File

@ -137,7 +137,7 @@ if __name__ == "__main__":
parser.add_argument("--use_wandb", default=False, action="store_true") parser.add_argument("--use_wandb", default=False, action="store_true")
parser.add_argument("--wandb_project", type=str, default="MiniMind-V") parser.add_argument("--wandb_project", type=str, default="MiniMind-V")
parser.add_argument("--num_workers", type=int, default=8) 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("--images_path", type=str, default="./dataset/sft_images")
parser.add_argument("--ddp", action="store_true") parser.add_argument("--ddp", action="store_true")
parser.add_argument("--accumulation_steps", type=int, default=1) parser.add_argument("--accumulation_steps", type=int, default=1)