Change sent flags when running fp16 networks

2021-11-11 15:36:48 +02:00 · 2021-11-11 15:36:48 +02:00 · b4fa2381ae
parent 1792d5f8a4
commit b4fa2381ae
1 changed files with 4 additions and 0 deletions
--- a/mindspore/lite/test/st/scripts/run_net_train.sh
+++ b/mindspore/lite/test/st/scripts/run_net_train.sh
@ -360,6 +360,10 @@ function Run_arm() {
         echo "rm -f ${inference_file}*" >> ${run_arm_log_file}
            echo "rm -f ${inference_file}*" >> ${adb_cmd_run_file}
        fi
+        if [ ${enable_fp16} = "true" ]; then
+            export_file=""
+            inference_file=""
+        fi
        adb -s ${device_id} shell < ${adb_cmd_run_file} >> ${run_arm_log_file}
        adb_cmd=$(cat <<-ENDM
        export LD_LIBRARY_PATH=./:/data/local/tmp/:${tmp_dir};./benchmark_train \