change wide&deep train batchsize

This commit is contained in:
lizhenyu 2020-08-14 16:50:03 +08:00
parent 19c800a758
commit d409de0e3b
1 changed files with 1 additions and 1 deletions

View File

@ -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 &