提交 5ad28174 编写于 作者: X xiongkun 提交者: Wei Shengyu

add statuscheck and follow the tipc_standard

上级 c9f8b87e
......@@ -20,16 +20,19 @@ model_name=$(func_parser_value "${lines[1]}")
LOG_PATH="./test_tipc/output/${model_name}/${MODE}"
rm -rf $LOG_PATH
mkdir -p ${LOG_PATH}
status_log="${LOG_PATH}/results_python.log"
# start dygraph train
dygraph_output=$LOG_PATH/dygraph_output.txt
dygraph_output=$LOG_PATH/python_train_infer_dygraph_output.txt
dygraph_loss=$LOG_PATH/dygraph_loss.txt
sed -i '15ctrainer:norm_train' ${FILENAME}
cmd="bash test_tipc/test_train_inference_python.sh ${FILENAME} $MODE >$dygraph_output 2>&1"
echo $cmd
eval $cmd
# start dy2static train
dy2static_output=$LOG_PATH/dy2static_output.txt
dy2static_output=$LOG_PATH/python_train_infer_dy2static_output.txt
dy2static_loss=$LOG_PATH/dy2static_loss.txt
sed -i '15ctrainer:to_static_train' ${FILENAME}
cmd="bash test_tipc/test_train_inference_python.sh ${FILENAME} $MODE >$dy2static_output 2>&1"
echo $cmd
......@@ -37,21 +40,17 @@ eval $cmd
# analysis and compare the losses.
dyout=`cat $dy2static_output | python3 test_tipc/extract_loss.py -v 'Iter:' -e 'loss: {%f},'`
stout=`cat $dygraph_output | python3 test_tipc/extract_loss.py -v 'Iter:' -e 'loss: {%f},' `
echo $dyout
echo $stout
stout=`cat $dygraph_output | python3 test_tipc/extract_loss.py -v 'Iter:' -e 'loss: {%f},' `
echo $dyout > $dygraph_loss
echo $stout > $dy2static_loss
diff_log=$LOG_PATH/diff_log.txt
diff_cmd="diff -w $dygraph_loss $dy2static_loss | tee $diff_log"
eval $diff_cmd
last_status=$?
if [ "$dyout" = "" ]; then
echo "Failed to run model."
exit -1
status_check 2 $diff_cmd $status_log $model_name $diff_log
fi
if [ "$dyout" = "$stout" ]; then
echo "Successful Run Dy2static."
exit 0
else
echo "Loss is not equal."
echo "Dygraph Loss is: "
echo $dyout
echo "Dy2Static Loss is: "
echo $stout
exit -1
if [ "$stout" = "" ]; then
status_check 2 $diff_cmd $status_log $model_name $diff_log
fi
status_check $last_status $diff_cmd $status_log $model_name $diff_log
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册