reduce the training samples for infer model

PaddlePaddle · Dec 21, 2017 · 81e15bc · 81e15bc
1 parent 5c7fada
commit 81e15bc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/benchmark/paddle/image/run_mkl_infer.sh b/benchmark/paddle/image/run_mkl_infer.sh
@@ -37,7 +37,7 @@ function infer() {
       --trainer_count=1 \
       --num_passes=1 \
       --save_dir="models/${topology}-${layer_num}" \
-      --config_args="batch_size=128,layer_num=${layer_num}" \
+      --config_args="batch_size=128,layer_num=${layer_num},num_samples=256" \
       > /dev/null 2>&1
     echo "Done"
   fi