From e341491f2dd45d466064b819df95f28a033dd701 Mon Sep 17 00:00:00 2001 From: Faith Xu Date: Wed, 10 Nov 2021 15:41:11 -0800 Subject: [PATCH] [Docs] Fix tutorial links (#9721) * Remove incorrect sample links * Update tutorial list * Add link to BERT model acceleration tutorial --- docs/tutorials/accelerate-pytorch/pytorch.md | 7 +++++++ docs/tutorials/tensorflow.md | 6 +----- 2 files changed, 8 insertions(+), 5 deletions(-) diff --git a/docs/tutorials/accelerate-pytorch/pytorch.md b/docs/tutorials/accelerate-pytorch/pytorch.md index 79777eeb60..10c09f98bb 100644 --- a/docs/tutorials/accelerate-pytorch/pytorch.md +++ b/docs/tutorials/accelerate-pytorch/pytorch.md @@ -19,4 +19,11 @@ ONNX Runtime can be used to accelerate PyTorch models inferencing. ## Accelerate PyTorch model inferencing {: .no_toc } +### BERT +* [Accelerate BERT model on CPU](https://github.com/microsoft/onnxruntime/blob/master/onnxruntime/python/tools/transformers/notebooks/PyTorch_Bert-Squad_OnnxRuntime_CPU.ipynb) +* [Accelerate BERT model on GPU](https://github.com/microsoft/onnxruntime/blob/master/onnxruntime/python/tools/transformers/notebooks/PyTorch_Bert-Squad_OnnxRuntime_GPU.ipynb) * [Accelerate reduced size BERT model through quantization](https://github.com/microsoft/onnxruntime-inference-examples/blob/main/quantization/notebooks/bert/Bert-GLUE_OnnxRuntime_quantization.ipynb) + +### GPT-2 +* [Accelerate GPT2 on CPU](https://github.com/microsoft/onnxruntime/blob/master/onnxruntime/python/tools/transformers/notebooks/Inference_GPT2_with_OnnxRuntime_on_CPU.ipynb) +* [Accelerate GPT2 (with one step search) on CPU](https://github.com/microsoft/onnxruntime/blob/master/onnxruntime/python/tools/transformers/notebooks/Inference_GPT2-OneStepSearch_OnnxRuntime_CPU.ipynb) diff --git a/docs/tutorials/tensorflow.md b/docs/tutorials/tensorflow.md index 2ffc5f70bf..f4dab10b14 100644 --- a/docs/tutorials/tensorflow.md +++ b/docs/tutorials/tensorflow.md @@ -23,13 +23,9 @@ These examples use the [TensorFlow-ONNX converter](https://github.com/onnx/tenso * [TensorFlow: Image classification (efficientnet-edge)](https://github.com/onnx/tensorflow-onnx/blob/master/tutorials/efficientnet-edge.ipynb) * [TensorFlow: Image classification (efficientnet-lite)](https://github.com/onnx/tensorflow-onnx/blob/master/tutorials/efficientnet-lite.ipynb) * [TensorFlow: Natural Language Processing (BERT)](https://github.com/onnx/tensorflow-onnx/blob/master/tutorials/BertTutorial.ipynb) +* [TensorFlow: Accelerate BERT model](https://github.com/microsoft/onnxruntime/blob/master/onnxruntime/python/tools/transformers/notebooks/Tensorflow_Tf2onnx_Bert-Squad_OnnxRuntime_CPU.ipynb) * [Keras: Image classification (Resnet 50)](https://github.com/onnx/tensorflow-onnx/blob/master/tutorials/keras-resnet50.ipynb) ### TFLite {: .no_toc } * [TFLite: Image classifciation (mobiledet)](https://github.com/onnx/tensorflow-onnx/blob/master/tutorials/mobiledet-tflite.ipynb) - -## Accelerate TensorFlow model inferencing -{: .no_toc } -* [Accelerate BERT model on CPU](https://github.com/microsoft/onnxruntime/blob/master/onnxruntime/python/tools/transformers/notebooks/PyTorch_Bert-Squad_OnnxRuntime_CPU.ipynb) -* [Accelerate BERT model on GPU](https://github.com/microsoft/onnxruntime/blob/master/onnxruntime/python/tools/transformers/notebooks/PyTorch_Bert-Squad_OnnxRuntime_GPU.ipynb) \ No newline at end of file