Commit 6a3bcef8 authored by Hongkun Yu's avatar Hongkun Yu Committed by A. Unique TensorFlower
Browse files

Update transformer readme

PiperOrigin-RevId: 291503470
parent ca6f9026
......@@ -135,9 +135,7 @@ tensorboard --logdir=$MODEL_DIR
- --num_gpus=2+: Uses tf.distribute.MirroredStrategy to run synchronous
distributed training across the GPUs.
#### Using TPUs
Note: This model will **not** work with TPUs on Colab.
#### Using Cloud TPUs
You can train the Transformer model on Cloud TPUs using
`tf.distribute.TPUStrategy`. If you are not familiar with Cloud TPUs, it is
......
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