Skip to content

Commit 273c593

Browse files
benchmark llama worker=1 (#8305)
1 parent 277f45b commit 273c593

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/test_tipc/configs/llama/train_infer_python.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@ null:null
1313
null:null
1414
##
1515
trainer:norm_train
16-
norm_train:../llm/llama/benchmark.py --model_name_or_path facebook/llama-7b-2l --do_train --max_steps 500 --recompute False --overwrite_output_dir --output_dir ./checkpoints/ --fp16_opt_level O2 --learning_rate 3e-5 --warmup_steps 0 --seed 23 --logging_steps 1
16+
norm_train:../llm/llama/benchmark.py --model_name_or_path facebook/llama-7b-2l --do_train --max_steps 500 --recompute False --overwrite_output_dir --output_dir ./checkpoints/ --fp16_opt_level O2 --learning_rate 3e-5 --warmup_steps 0 --seed 23 --logging_steps 1 --dataloader_num_workers 1
1717
pact_train:null
1818
fpgm_train:null
1919
distill_train:null

0 commit comments

Comments
 (0)