onnxruntime/samples/python/pytorch_transformer

TransformerModel example

This example was adapted from Pytorch's Sequence-to-Sequence Modeling with nn.Transformer and TorchText tutorial

Requirements

• PyTorch 1.6+
• TorchText 0.6+
• ONNX Runtime 1.5+

Running PyTorch version

python pt_model.py

Running ONNX Runtime version

python ort_model.py