torchort 0.0.7.dev20210315

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