This commit is contained in:
hailin 2025-09-08 14:47:00 +08:00
parent a820de7aaa
commit b433fd0525
1 changed files with 2 additions and 2 deletions

View File

@ -28,9 +28,9 @@ export OPENBLAS_NUM_THREADS=8
deepspeed --hostfile hostfile \
--num_nodes 6 --num_gpus 4 \
/home/test/jd_train/train_sft_ds.py \
--model_name_or_path /home/test/Qwen3-1.7B \
--model_name_or_path /home/test/Qwen3-32B \
--data_glob "/home/test/datasets/my_corpus/train.jsonl" \
--output_dir /home/test/checkpoints/q3-1.7b-ds4 \
--output_dir /home/test/checkpoints/q3-32b-ds4 \
--seq_len 512 \
--per_device_train_batch_size 1 \
--gradient_accumulation_steps 1 \