|
|
@@ -98,10 +98,10 @@
|
|
|
"* [Information about NVIDIA Tensor Cores](https://developer.nvidia.com/tensor-cores)\n",
|
|
|
"* [Post on TensorFlow blog explaining Automatic Mixed Precision](https://medium.com/tensorflow/automatic-mixed-precision-in-tensorflow-for-faster-ai-training-on-nvidia-gpus-6033234b2540)\n",
|
|
|
"\n",
|
|
|
- "**TensorFlow Automatic Mixed-Precision API",
|
|
|
- "The method presented in this notebook is the API used in TensorFlow 1.14 and newer: tf.train.experimental.enable_mixed_precision_graph_rewrite()",
|
|
|
- "Switch to using mixed precision by simply wrapping a tf.keras.optimizers Optimizer in tf.train.experimental.enable_mixed_precision_graph_rewrite()",
|
|
|
- "More details: [https://www.tensorflow.org/api_docs/python/tf/train/experimental/enable_mixed_precision_graph_rewrite](https://www.tensorflow.org/api_docs/python/tf/train/experimental/enable_mixed_precision_graph_rewrite)"
|
|
|
+ "**TensorFlow Automatic Mixed-Precision API\n",
|
|
|
+ "The method presented in this notebook is the API used in TensorFlow 1.14 and newer: tf.train.experimental.enable_mixed_precision_graph_rewrite()\n",
|
|
|
+ "Switch to using mixed precision by simply wrapping a tf.keras.optimizers Optimizer in tf.train.experimental.enable_mixed_precision_graph_rewrite()\n",
|
|
|
+ "More details: [https://www.tensorflow.org/api_docs/python/tf/train/experimental/enable_mixed_precision_graph_rewrite](https://www.tensorflow.org/api_docs/python/tf/train/experimental/enable_mixed_precision_graph_rewrite)\n"
|
|
|
]
|
|
|
},
|
|
|
{
|