Skip to content

GitLab

Commit 0b87b8b0 authored by A. Unique TensorFlower's avatar A. Unique TensorFlower
Browse files

Merge pull request #7978 from artitw:artitw-distribution-strategy 

PiperOrigin-RevId: 287584771
parents 3dee3b86 bca18d0b
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
official/nlp/bert/README.md
View file @ 0b87b8b0
...@@ -239,7 +239,7 @@ python run_classifier.py \ ...@@ -239,7 +239,7 @@ python run_classifier.py \
--learning_rate=2e-5 \ --learning_rate=2e-5 \
--num_train_epochs=3 \ --num_train_epochs=3 \
--model_dir=${MODEL_DIR} \ --model_dir=${MODEL_DIR} \
--distribution_strategy=mirror --distribution_strategy=mirrored
``` ```
To use TPU, you only need to switch distribution strategy type to `tpu` with TPU To use TPU, you only need to switch distribution strategy type to `tpu` with TPU
...@@ -295,7 +295,7 @@ python run_squad.py \ ...@@ -295,7 +295,7 @@ python run_squad.py \
--learning_rate=8e-5 \ --learning_rate=8e-5 \
--num_train_epochs=2 \ --num_train_epochs=2 \
--model_dir=${MODEL_DIR} \ --model_dir=${MODEL_DIR} \
--distribution_strategy=mirror --distribution_strategy=mirrored
``` ```
To use TPU, you need switch distribution strategy type to `tpu` with TPU To use TPU, you need switch distribution strategy type to `tpu` with TPU
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment