Upload math_train.sh with huggingface_hub
Browse files- math_train.sh +2 -2
math_train.sh
CHANGED
|
@@ -4,9 +4,9 @@ VLLM_USE_V1=1 ASCEND_RT_VISIBLE_DEVICES=0 python main.py --mode train \
|
|
| 4 |
--train_data processed_data/MATH/train.jsonl \
|
| 5 |
--val_data processed_data/MATH/val.jsonl \
|
| 6 |
--num_epochs 5 \
|
| 7 |
-
--batch_size
|
| 8 |
--num_samples 4 \
|
| 9 |
--token_budget 4096 \
|
| 10 |
--checkpoint_dir checkpoints/MATH/pangu \
|
| 11 |
--task_type math \
|
| 12 |
-
|
|
|
|
| 4 |
--train_data processed_data/MATH/train.jsonl \
|
| 5 |
--val_data processed_data/MATH/val.jsonl \
|
| 6 |
--num_epochs 5 \
|
| 7 |
+
--batch_size 16 \
|
| 8 |
--num_samples 4 \
|
| 9 |
--token_budget 4096 \
|
| 10 |
--checkpoint_dir checkpoints/MATH/pangu \
|
| 11 |
--task_type math \
|
| 12 |
+
|