fix_log_path
parent
953a99fa4c
commit
20c04f61b8
benchmark
|
@ -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)
|
||||
|
|
|
@ -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 (必填)
|
||||
|
|
Loading…
Reference in New Issue