diff --git a/test_tipc/test_train_dy2static_python.sh b/test_tipc/test_train_dy2static_python.sh index a4a990d72f66ee1a7706360f9e4fb5a363576725..13f930aac1dcc2ee48ba2373aa37bb66af4a173a 100644 --- a/test_tipc/test_train_dy2static_python.sh +++ b/test_tipc/test_train_dy2static_python.sh @@ -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