diff --git a/benchmark/run_all.sh b/benchmark/run_all.sh index 8b03498b5a94052659b0f6b4838d270f457c5075..648d89783ddda2f4ddfa2eb5553373ed5623698c 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 606235d8f22d849131bef44bb8c705efb4f1e252..e05780deaca813bd50d8ad9385deb089fc1ed4c9 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)