add wide_deeep gpu

BamLubi · Aug 10, 2022 · 1966218 · 1966218
1 parent f265eaa
commit 1966218
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 4 deletions.
diff --git a/test_tipc/benchmark_train.sh b/test_tipc/benchmark_train.sh
@@ -152,7 +152,7 @@ else
     precision=${params_list[2]}
     # run_process_type=${params_list[3]}
     run_mode=${params_list[3]}
-    func_sed_params "$FILENAME" "${line_runmode}" "$MODE=$run_mode"
+    func_sed_params "$FILENAME" "${line_runmode}" "$run_mode"
     device_num=${params_list[4]}
     IFS=";"
 

diff --git a/test_tipc/configs/dnn/train_infer_python.txt b/test_tipc/configs/dnn/train_infer_python.txt
@@ -56,5 +56,5 @@ epoch:50
 run_mode:PSGPU
 fp_items:null
 device_num:N1C8
-gpu_config:models/rank/wide_deep/config_gpubox.yaml
-cpu_config:models/rank/wide_deep/config_cpups.yaml
+gpu_config:models/rank/dnn/config_gpubox.yaml
+cpu_config:models/rank/dnn/config_cpups.yaml
diff --git a/test_tipc/test_train_inference_python.sh b/test_tipc/test_train_inference_python.sh
@@ -224,7 +224,9 @@ if [ ${MODE} = "benchmark_train" ]; then
 	  mkdir ./log
 	  echo "Create log floder for store running log"
 	fi
-
+    echo "aaaaaaaaaaaa"
+    echo ${run_mode_value}
+    echo "bbbbbbbbbbbbbb"
 	if [ ${run_mode_value} = "PSGPU" ]; then
         export FLAGS_LAUNCH_BARRIER=0
         export PADDLE_TRAINER_ID=0