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 &