File size: 395 Bytes
1482463 3c7660e 1482463 3c7660e |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
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 \
|