Skip to content

Commit f6dae32

Browse files
committed
[ONNX 3] Add ONNX registry tutorial
1 parent 9867f0e commit f6dae32

11 files changed

+458
-1
lines changed
Loading
7.37 KB
Loading
12.4 KB
Loading
8.41 KB
Loading
22.1 KB
Loading
6.8 KB
Loading

beginner_source/onnx/README.txt

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,3 +8,7 @@ ONNX
88
2. export_simple_model_to_onnx_tutorial.py
99
Export a PyTorch model to ONNX
1010
https://pytorch.org/tutorials/beginner/onnx/export_simple_model_to_onnx_tutorial.html
11+
12+
3. onnx_registry_tutorial.py
13+
Introduction to ONNX
14+
https://pytorch.org/tutorials/beginner/onnx/onnx_registry_tutorial.html

beginner_source/onnx/intro_onnx.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -32,13 +32,21 @@
3232
Dependencies
3333
------------
3434
35+
PyTorch 2.1.0 or newer is required.
36+
3537
The ONNX exporter depends on extra Python packages:
3638
3739
- `ONNX <https://onnx.ai>`_
3840
- `ONNX Script <https://onnxscript.ai>`_
3941
4042
They can be installed through `pip <https://pypi.org/project/pip/>`_:
4143
44+
.. note::
45+
This tutorial leverages `onnxscript <https://github.com/microsoft/onnxscript#readme>`__
46+
to create custom ONNX operators. onnxscript is a Python library that allows users to
47+
create custom ONNX operators in Python. It is a prerequisite learning material for
48+
this tutorial. Please make sure you have read the onnxscript tutorial before proceeding.
49+
4250
.. code-block:: bash
4351
4452
pip install --upgrade onnx onnxscript

0 commit comments

Comments
 (0)