Metadata-Version: 2.1
Name: torchort
Version: 0.0.7.dev20210317
Summary: training Pytorch models with onnxruntime
Home-page: https://github.com/pytorch/ort
Author: torchort contributors
License: UNKNOWN
Project-URL: Bug Tracker, https://github.com/pytorch/ort/issues
Platform: UNKNOWN
Classifier: Programming Language :: Python :: 3
Classifier: License :: OSI Approved :: MIT License
Classifier: Operating System :: OS Independent
Requires-Python: >=3.6
Description-Content-Type: text/markdown

# 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:
    (uninstall onnxruntime version that do not support gpu and training)
    pip install torchort onnxruntime=1.9.0+cu111_training

### nightly:
    (make sure you are not in the ort repo folder - otherwise torchort is taken as already installed)
    pip install --pre torchort ort-gpu-nightly-training

    (eventually we are aiming at: pip install --pre torchort onnxruntime=1.9.0+cu111_training)

## to test:
    python ./ort/tests/bert_for_sequence_classification.py
## to use torchort within PyTorch training scripts:
    import onnxruntime
    from torchort import ORTModule
    model = ORTModule(model)
    # normal PyTorch training script follows


