From 890637c175e2c0150e92bd3f4432081aebf264cc Mon Sep 17 00:00:00 2001 From: JamesLim-sy Date: Thu, 11 Aug 2022 07:08:25 +0000 Subject: [PATCH] max_iter changes for transformer base model benchmark test --- scripts/dynamic_graph_models.sh | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/scripts/dynamic_graph_models.sh b/scripts/dynamic_graph_models.sh index 9c6ee97ced..4445cb8d82 100644 --- a/scripts/dynamic_graph_models.sh +++ b/scripts/dynamic_graph_models.sh @@ -271,10 +271,10 @@ dy_transformer(){ do model_name="transformer_${mode_item}_${fp_item}" echo "index is speed, ${model_name} 1gpu begin" - CUDA_VISIBLE_DEVICES=5 bash run_benchmark.sh 1 sp 600 ${mode_item} ${fp_item} | tee ${log_path}/dynamic_${model_name}_speed_1gpus 2>&1 + CUDA_VISIBLE_DEVICES=5 bash run_benchmark.sh 1 sp 5000 ${mode_item} ${fp_item} | tee ${log_path}/dynamic_${model_name}_speed_1gpus 2>&1 sleep 60 echo "index is speed, ${model_name} 8gpus begin, mp" - CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 bash run_benchmark.sh 1 mp 500 ${mode_item} ${fp_item} | tee ${log_path}/dynamic_${model_name}_speed_8gpus 2>&1 + CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 bash run_benchmark.sh 1 mp 5000 ${mode_item} ${fp_item} | tee ${log_path}/dynamic_${model_name}_speed_8gpus 2>&1 done done }