Browse Source

change wide&deep train batchsize

tags/v0.7.0-beta
lizhenyu 5 years ago
parent
commit
d409de0e3b
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      model_zoo/official/recommend/wide_and_deep/script/run_multigpu_train.sh

+ 1
- 1
model_zoo/official/recommend/wide_and_deep/script/run_multigpu_train.sh View File

@@ -25,5 +25,5 @@ mpirun --allow-run-as-root -n $RANK_SIZE \
python -s ${self_path}/../train_and_eval_distribute.py \ python -s ${self_path}/../train_and_eval_distribute.py \
--device_target="GPU" \ --device_target="GPU" \
--data_path=$DATASET \ --data_path=$DATASET \
--batch_size=8000 \
--batch_size=16000 \
--epochs=$EPOCH_SIZE > log.txt 2>&1 & --epochs=$EPOCH_SIZE > log.txt 2>&1 &

Loading…
Cancel
Save