From d409de0e3bd1d79f709515c6112c655ce0f3d5d5 Mon Sep 17 00:00:00 2001
From: lizhenyu <lizhenyu13@huawei.com>
Date: Fri, 14 Aug 2020 16:50:03 +0800
Subject: [PATCH] change wide&deep train batchsize

---
 .../recommend/wide_and_deep/script/run_multigpu_train.sh        | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/model_zoo/official/recommend/wide_and_deep/script/run_multigpu_train.sh b/model_zoo/official/recommend/wide_and_deep/script/run_multigpu_train.sh
index e0e08ab80e..ae8d31e36e 100644
--- a/model_zoo/official/recommend/wide_and_deep/script/run_multigpu_train.sh
+++ b/model_zoo/official/recommend/wide_and_deep/script/run_multigpu_train.sh
@@ -25,5 +25,5 @@ mpirun --allow-run-as-root -n $RANK_SIZE                    \
     python -s ${self_path}/../train_and_eval_distribute.py  \
         --device_target="GPU"                               \
         --data_path=$DATASET                                \
-        --batch_size=8000                                   \
+        --batch_size=16000                                  \
         --epochs=$EPOCH_SIZE > log.txt 2>&1 &