onnxruntime/orttraining/orttraining/python
liqunfu 6260d073b3
Glue parallel training (#4550)
add mpi size, rank python API

add single node parallel training example
2020-08-21 21:24:27 -07:00
..
experimental Update LRScheduler to use scheduling similar to HuggingFace (#4880) 2020-08-21 10:24:04 -07:00
training Add new PytTrch front-end (#4815) 2020-08-17 09:45:25 -07:00
checkpointing_utils.py Add Distributed Checkpointing support (#3639) 2020-04-29 14:52:21 -07:00
ort_trainer.py Add new PytTrch front-end (#4815) 2020-08-17 09:45:25 -07:00
orttraining_pybind_state.cc Glue parallel training (#4550) 2020-08-21 21:24:27 -07:00
pt_patch.py Remove 'model_.' prefix from onnx model initializers in training (#3881) 2020-05-20 10:06:31 -07:00