torch-ort-poc 0.0.6.dev20210314

training Pytorch models with onnxruntime

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-poc

nightly:

pip install --pre torch-ort-poc

to use torch_ort within PyTorch training scripts:

import onnxruntime
from torch_ort import ORTModule
model = ORTModule(model)
# normal PyTorch training script follows