feat: 添加MOELORA支持,优化训练和评估脚本,修复拼写错误,提升代码可读性

This commit is contained in:
2025-06-03 20:25:20 +08:00
parent b84ebb03c7
commit d686cbc254
11 changed files with 221 additions and 138 deletions
+15
View File
@@ -0,0 +1,15 @@
#!/bin/bash
accelerate launch --config_file configs/accelerate_configs/deepspeed_zero1.yaml evaluation.py \
--dataset_name textvqa \
--use_peft \
--peft_type MOELORA \
--model_name_or_path Qwen/Qwen2.5-Omni-3B \
--lora_target_modules .*model\.layers.*proj\|.*merger.*0\|.*merger.*1 \
--per_device_train_batch_size 3 \
--per_device_eval_batch_size 2 \
--gradient_accumulation_steps 2 \
--output_dir ./checkpoint/qwen2_5omni_moelora/ \
--bf16 \
--torch_dtype bfloat16
# --eval_strategy epoch \
+3 -2
View File
@@ -18,7 +18,8 @@ accelerate launch --config_file configs/accelerate_configs/deepspeed_zero1.yaml
--lr_scheduler_type cosine \
--bf16 \
--torch_dtype bfloat16 \
--logging_steps 10 \
--logging_steps 100 \
--gradient_checkpointing \
--weight_decay 0.1 \
# --resume_from_checkpoint /root/autodl-tmp/zhouyunyao/projects/CL-LMM/src/checkpoint/qwen2_alllinear/checkpoint-1000
--eval_strategy steps \
# --resume_from_checkpoint /root/autodl-tmp/zhouyunyao/projects/CL-LMM/src/checkpoint/qwen2_5omni_moelora/checkpoint-1500