fix dnn infer

shenghuitan · Jan 26, 2021 · fa0f538 · fa0f538
1 parent ebdda14
commit fa0f538
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/models/rank/deepfm/benchmark.yaml b/models/rank/deepfm/benchmark.yaml
@@ -41,7 +41,7 @@ runner:
   train_reader_path: "benchmark_reader"
   model_save_path: "model"
 
-  infer_batch_size: 2
+  infer_batch_size: 1000
   test_data_dir: "test_data"
   infer_reader_path: "benchmark_reader"
   infer_load_path: "model"

diff --git a/models/rank/dnn/benchmark.yaml b/models/rank/dnn/benchmark.yaml
@@ -43,7 +43,7 @@ runner:
   train_reader_path: "benchmark_reader"
   model_save_path: "model"
 
-  infer_batch_size: 2
+  infer_batch_size: 1000
   test_data_dir: "test_data"
   infer_reader_path: "benchmark_reader"
   infer_load_path: "model"

diff --git a/models/rank/wide_deep/benchmark.yaml b/models/rank/wide_deep/benchmark.yaml
@@ -41,7 +41,7 @@ runner:
   train_reader_path: "benchmark_reader"
   model_save_path: "model"
 
-  infer_batch_size: 2
+  infer_batch_size: 1000
   test_data_dir: "test_data"
   infer_reader_path: "benchmark_reader"
   infer_load_path: "model"