onnxruntime/orttraining/orttraining/python
Bowen Bao 15cb4b3023
Fix session load state & run extra_postpasses only once (#4255)
* Fix session load state & run extra_postpasses only once

* add testcase for onnx model as well
2020-06-23 11:45:26 -07:00
..
training
checkpointing_utils.py Add Distributed Checkpointing support (#3639) 2020-04-29 14:52:21 -07:00
ort_trainer.py Fix session load state & run extra_postpasses only once (#4255) 2020-06-23 11:45:26 -07:00
orttraining_pybind_state.cc Enable ORT with CUDA 11 toolkit (#4168) 2020-06-15 08:47:03 -07:00
postprocess.py Add ONNX Training Post-Passes to Front-End - Cont (#4041) 2020-06-15 10:33:26 -07:00
pt_patch.py Remove 'model_.' prefix from onnx model initializers in training (#3881) 2020-05-20 10:06:31 -07:00