diff --git a/benchmark/run_all.sh b/benchmark/run_all.sh index 3c6cee83d061c8950bec5a78b6a1db410d08aa88..a6d6884171d2ce4db324c9fe64f5cb077da3da16 100644 --- a/benchmark/run_all.sh +++ b/benchmark/run_all.sh @@ -5,6 +5,7 @@ # pip install ... # 2 拷贝该模型需要数据、预训练模型 # 3 批量运行(如不方便批量,1,2需放到单个模型中) +log_path=${LOG_PATH_INDEX_DIR:-$(pwd)} # LOG_PATH_INDEX_DIR 后续QA设置参数 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 e05780deaca813bd50d8ad9385deb089fc1ed4c9..7c1513e608b1bcce217e9c767718f685a1051125 100644 --- a/benchmark/run_benchmark.sh +++ b/benchmark/run_benchmark.sh @@ -8,7 +8,7 @@ function _set_params(){ fp_item=${3:-"fp32"} # fp32|fp16 epochs=${4:-"2"} # 可选,如果需要修改代码提前中断 model_name=${5:-"model_name"} - run_log_path="${TRAIN_LOG_DIR:-$(pwd)}/benchmark" # TRAIN_LOG_DIR 后续QA设置该参数 + run_log_path=${TRAIN_LOG_DIR:-$(pwd)} # TRAIN_LOG_DIR 后续QA设置该参数 index=1 mission_name="图像分类" # 模型所属任务名称,具体可参考scripts/config.ini (必填)