From da3a7af43b331fc2d3c0100d6a59914bef8f90d2 Mon Sep 17 00:00:00 2001
From: hysunflower <nothing_upc@163.com>
Date: Tue, 23 Nov 2021 12:10:40 +0000
Subject: [PATCH] update_scripts_benchmark

---
 benchmark/run_all.sh       | 1 -
 benchmark/run_benchmark.sh | 4 ++--
 2 files changed, 2 insertions(+), 3 deletions(-)

diff --git a/benchmark/run_all.sh b/benchmark/run_all.sh
index 8b03498b..648d8978 100644
--- a/benchmark/run_all.sh
+++ b/benchmark/run_all.sh
@@ -5,7 +5,6 @@
 # pip install ...
 # 2 拷贝该模型需要数据、预训练模型
 # 3 批量运行（如不方便批量，1，2需放到单个模型中）
-log_path=./benchmark
 model_mode_list=(MobileNetV1 MobileNetV2 MobileNetV3_large_x1_0 ShuffleNetV2_x1_0 HRNet_W48_C SwinTransformer_tiny_patch4_window7_224 alt_gvt_base)    # benchmark 监控模型列表
 #model_mode_list=(MobileNetV1 MobileNetV2 MobileNetV3_large_x1_0 EfficientNetB0 ShuffleNetV2_x1_0 DenseNet121 HRNet_W48_C SwinTransformer_tiny_patch4_window7_224 alt_gvt_base)   # 该脚本支持列表
 fp_item_list=(fp32)
diff --git a/benchmark/run_benchmark.sh b/benchmark/run_benchmark.sh
index 606235d8..e05780de 100644
--- a/benchmark/run_benchmark.sh
+++ b/benchmark/run_benchmark.sh
@@ -1,5 +1,5 @@
 #!/usr/bin/env bash
-set -x
+set -xe
 # 运行示例：CUDA_VISIBLE_DEVICES=0 bash run_benchmark.sh ${run_mode} ${bs_item} ${fp_item} 500 ${model_mode}
 # 参数说明
 function _set_params(){
@@ -35,7 +35,7 @@ function _train(){
         model_config=`find ppcls/configs/ImageNet -name ${model_name}_fp16.yaml` 
     fi
 
-    train_cmd="-c ${model_config} -o DataLoader.Train.sampler.batch_size=${batch_size} -o Global.epochs=${epochs}  -o Global.eval_during_train=False"
+    train_cmd="-c ${model_config} -o DataLoader.Train.sampler.batch_size=${batch_size} -o Global.epochs=${epochs}  -o Global.eval_during_train=False -o Global.print_batch_step=2"
     case ${run_mode} in
     sp) train_cmd="python -u tools/train.py ${train_cmd}" ;;
     mp)
-- 
GitLab