onnxruntime/orttraining/orttraining/python/training
Wei-Sheng Chin 7df8f84228
Improve DORT document (#13790)
1. Refine words based on PyTorch changes.
2. Make the need of inference mode clearer. A test is added.
2022-11-30 16:55:25 -08:00
..
amp
api Training API to export the eval model to an inference model (#13345) 2022-10-27 09:34:01 -07:00
experimental
onnxblock Fix on-device training ExportModelForInferencing api (#13510) 2022-10-31 21:29:06 -07:00
optim [ORTModule] Update Supported DeepSpeed Version for FP16_Optimizer (#13305) 2022-10-13 13:03:01 +08:00
ortmodule Move a function call outside a loop in ORTModule (#13771) 2022-11-30 12:49:41 +01:00
torchdynamo Improve DORT document (#13790) 2022-11-30 16:55:25 -08:00
utils/data
__init__.py
_checkpoint_storage.py
_utils.py replace 'master' branch ref to 'main' for onnx repo (#12678) 2022-08-30 13:41:42 -07:00
checkpoint.py
model_desc_validation.py
orttrainer.py Deprecate ORTTrainer (#13022) 2022-09-23 18:10:09 -07:00
orttrainer_options.py
postprocess.py