fix tipc lite_train_whole_infer bug

pull/1594/head
dongshuilong 2021-12-24 03:11:26 +00:00
parent a50b3e9fc2
commit f6f2a20126
2 changed files with 4 additions and 4 deletions

View File

@ -118,9 +118,7 @@ if [ ${MODE} = "lite_train_lite_infer" ] || [ ${MODE} = "lite_train_whole_infer"
cd ILSVRC2012 cd ILSVRC2012
mv train.txt train_list.txt mv train.txt train_list.txt
mv val.txt val_list.txt mv val.txt val_list.txt
if [ ${MODE} = "lite_train_lite_infer" ];then cp -r train/* val/
cp -r train/* val/
fi
cd ../../ cd ../../
elif [ ${MODE} = "whole_infer" ] || [ ${MODE} = "klquant_whole_infer" ];then elif [ ${MODE} = "whole_infer" ] || [ ${MODE} = "klquant_whole_infer" ];then
# download data # download data

View File

@ -87,7 +87,9 @@ benchmark_key=$(func_parser_key "${lines[49]}")
benchmark_value=$(func_parser_value "${lines[49]}") benchmark_value=$(func_parser_value "${lines[49]}")
infer_key1=$(func_parser_key "${lines[50]}") infer_key1=$(func_parser_key "${lines[50]}")
infer_value1=$(func_parser_value "${lines[50]}") infer_value1=$(func_parser_value "${lines[50]}")
if [ ! $epoch_num ]; then
epoch_num=2
fi
LOG_PATH="./test_tipc/output" LOG_PATH="./test_tipc/output"
mkdir -p ${LOG_PATH} mkdir -p ${LOG_PATH}
status_log="${LOG_PATH}/results_python.log" status_log="${LOG_PATH}/results_python.log"