VLLM_USE_V1=1 ASCEND_RT_VISIBLE_DEVICES=0 python main.py --mode train \
  --device npu \
  --model_name "FreedomIntelligence/openPangu-Embedded-7B" \
  --train_data processed_data/MATH/train.jsonl \
  --val_data processed_data/MATH/val.jsonl \
  --num_epochs 5 \
  --batch_size 16 \
  --num_samples 4 \
  --token_budget 4096 \
  --checkpoint_dir checkpoints/MATH/pangu \
  --task_type math \