This commit is contained in:
parent
a820de7aaa
commit
b433fd0525
|
|
@ -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 \
|
||||
|
|
|
|||
Loading…
Reference in New Issue