saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-15 18:23:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
onnxruntime/docs/get-started/training-pytorch.md
Faith Xu bd9b15fd4d
[Docs] Docs updates (#14824) 
### Description
Staged: https://faxu.github.io/onnxruntime/docs/

Changes:
- ~~New dedicated section on sidebar for
[Mobile](https://faxu.github.io/onnxruntime/docs/mobile/) and
[Web](https://faxu.github.io/onnxruntime/docs/web/)~~ New dedicated
pages for
[Mobile](https://faxu.github.io/onnxruntime/docs/get-started/with-mobile.html)
and
[Web](https://faxu.github.io/onnxruntime/docs/get-started/with-web.html)
under Getting Started section
- [Transformers
optimizer](https://faxu.github.io/onnxruntime/docs/performance/transformers-optimization.html)
documentation
- Move contrib op section from "Custom Operators" page to ["Contrib
operators"
page](https://faxu.github.io/onnxruntime/docs/reference/operators/ContribOperators.html)
- Update [releases
page](https://faxu.github.io/onnxruntime/docs/reference/releases-servicing.html)
2023-03-08 11:54:58 -08:00

865 B
Raw Blame History

title parent nav_order
ORT Training with PyTorch Get Started 12

Get started with ORT for Training API (PyTorch)

{: .no_toc } The ORT Training API is a PyTorch frontend that implements the torch.nn.Module interface.

ORT Training Example

In this example we will go over how to use ORT for Training a model with PyTorch.

pip install torch-ort
python -m torch_ort.configure

Note: This installs the default version of the torch-ort and onnxruntime-training packages that are mapped to specific versions of the CUDA libraries. Refer to the install options in ONNXRUNTIME.ai.

  • Add ORTModule in the train.py
   from torch_ort import ORTModule
   .
   .
   .
   model = ORTModule(model)

Samples

ONNX Runtime Training Examples