!4458 change wide&deep training batchsize

Merge pull request !4458 from zyli2020/change_wide_and_deep_model_zoo_script
5 years ago · 6fd5323356
parent 50e78118fb d409de0e3b
commit 6fd5323356
1 changed files with 1 additions and 1 deletions
--- 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 &