fix tipc lite_train_whole_infer bug
parent
a50b3e9fc2
commit
f6f2a20126
|
@ -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
|
||||||
|
|
|
@ -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"
|
||||||
|
|
Loading…
Reference in New Issue