Skip to content

Commit a4561f4

Browse files
authored
Update link to tf.keras.mixed_precision example and Transformer in Mixed prevision guide
1 parent a224e08 commit a4561f4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

site/en/guide/mixed_precision.ipynb

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -819,7 +819,7 @@
819819
"* Double the training batch size if it does not reduce evaluation accuracy\n",
820820
"* On GPUs, ensure most tensor dimensions are a multiple of $8$ to maximize performance\n",
821821
"\n",
822-
"For more examples of mixed precision using the `tf.keras.mixed_precision` API, check the [official models repository](https://github.com/tensorflow/models/tree/master/official). Most official models, such as [ResNet](https://github.com/tensorflow/models/tree/master/official/legacy/image_classification) and [Transformer](https://github.com/tensorflow/models/tree/master/official/legacy/transformer), will run using mixed precision by passing `--dtype=fp16`.\n"
822+
"For an example of mixed precision using the `tf.keras.mixed_precision` API, check [functions and classes related to training performance](https://github.com/tensorflow/models/blob/master/official/modeling/performance.py). Check out the official models, such as [Transformer](https://github.com/tensorflow/models/blob/master/official/nlp/modeling/layers/transformer_encoder_block.py) for details.\n"
823823
]
824824
}
825825
],

0 commit comments

Comments
 (0)