Tutorial

This module was started to experiment around function torch.export.export() and see what kind of issues occur when leveraging that function to convert a torch.nn.Module into ONNX. The tutorial is a collection of examples or benchmark around that topic. Section Design explains how a converter works from torch model to the onnx graph. The official exporter is implemented in pytorch itself through the function torch.onnx.export().

• ShapeBuilder
• to_onnx: another export to investigate
• Unexpected Errors
  • Issues with torch.export.export
  • RuntimeError: CUDA error: CUDA-capable device(s) is/are busy or unavailable
• Start from a docker
  • Benchmark a llama model