python train.py \
    --model_name_or_path "/root/models/DeepSeek-R1-Distill-Qwen-1.5B" \
    --train_file "./data/train.jsonl" \
    --eval_file "./data/eval.jsonl" \
    --epochs 3 \
    --model_dir "./output_model" \
    --save_merged_model \
    --early_stopping_patience 3