提交 b9f7df5c 编写于 作者: D dongshuilong

fix bugs

上级 c1d3b5e3
......@@ -5,7 +5,7 @@ model_name_fpgm:ResNet50_vd|MobileNetV3_large_x1_0
model_name_kl:ResNet50_vd|MobileNetV3_large_x1_0
python:python3.7
gpu_list:0|0,1|-1
Global.epoch_num:10
Global.epochs:10
Global.output_dir:./output/
Global.pretrained_model:null
Global.save_inference_dir:null
......
......@@ -143,7 +143,7 @@ for gpu in ${gpu_list[*]}; do
for model_name in ${model_name_list[*]}; do
# not set epoch when whole_train_infer
if [ ${MODE} != "whole_train_infer" ]; then
set_epoch="-o ${epoch_key}=${epoch_num}"
set_epoch="-o ${epoch_key}=${epoch_value}"
else
set_epoch=" "
fi
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册