From 6e86b5aa1e52bb078b8b5ae024e91a25f6a5d470 Mon Sep 17 00:00:00 2001 From: Guillaume Klein Date: Fri, 12 Mar 2021 17:41:45 +0100 Subject: [PATCH] Update mixed precision mention in main README --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index 8a1ed6e55..b5210e17e 100644 --- a/README.md +++ b/README.md @@ -55,7 +55,7 @@ OpenNMT-tf is fully integrated in the TensorFlow 2 ecosystem: * Reusable layers extending [`tf.keras.layers.Layer`](https://www.tensorflow.org/api_docs/python/tf/keras/layers/Layer) * Multi-GPU training with [`tf.distribute`](https://www.tensorflow.org/api_docs/python/tf/distribute) and distributed training with [Horovod](https://github.com/horovod/horovod) -* Mixed precision support via a [graph optimization pass](https://www.tensorflow.org/api_docs/python/tf/train/experimental/enable_mixed_precision_graph_rewrite) +* Mixed precision training with [`tf.keras.mixed_precision`](https://www.tensorflow.org/guide/mixed_precision) * Visualization with [TensorBoard](https://www.tensorflow.org/tensorboard) * `tf.function` graph tracing that can be [exported to a SavedModel](https://opennmt.net/OpenNMT-tf/serving.html) and served with [TensorFlow Serving](https://github.com/OpenNMT/OpenNMT-tf/tree/master/examples/serving/tensorflow_serving) or [Python](https://github.com/OpenNMT/OpenNMT-tf/tree/master/examples/serving/python)