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