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 torchort

nightly:

pip install --pre torchort

to use torchort within PyTorch training scripts:

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