training Pytorch models with onnxruntime
Project description
training pytorch models with onnxruntime
to build (you need to update version number in version.txt in order to be able to upload python whl):
rm dist/*
python setup.py bdist_wheel
to publish:
twine upload dist/*
to install:
stable:
pip install torch-ort
nightly:
pip install --pre torch-ort
to use torch_ort within PyTorch training scripts:
import onnxruntime
from torch_ort import ORTModule
model = ORTModule(model)
# normal PyTorch training script follows
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distributions
No source distribution files available for this release.See tutorial on generating distribution archives.
Built Distribution
File details
Details for the file torch_ort-0.0.7.dev20210315-py3-none-any.whl.
File metadata
- Download URL: torch_ort-0.0.7.dev20210315-py3-none-any.whl
- Upload date:
- Size: 2.6 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/3.3.0 pkginfo/1.7.0 requests/2.22.0 setuptools/45.2.0 requests-toolbelt/0.9.1 tqdm/4.59.0 CPython/3.8.5
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | bcf22e9ba363708f57704e02593224fe9e1202a3fce926a643775796bd114f0e |
|
| MD5 | f3f9092c64015f591abd6c49b7c21a9e |
|
| BLAKE2b-256 | 3664e3b5792e7e3789528eb864aeaee64471cf242fd79e4b4e1444f6d4dda93c |
