Skip to content

GitLab

"examples/vscode:/vscode.git/clone" did not exist on "e619db24becce564a3646cef35ae83b06955867f"
Unverified Commit bca18d0b authored by Artit 'Art' Wangperawong's avatar Artit 'Art' Wangperawong Committed by GitHub
Browse files

fix distribution strategy

parent 01a51ee2
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
official/nlp/bert/README.md
View file @ bca18d0b
...@@ -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