onnxruntime/orttraining/orttraining/python/training
2023-05-02 16:58:16 -07:00
..
amp
api Python documentation for onnxruntime-training (#15765) 2023-05-02 16:58:16 -07:00
experimental
onnxblock Python documentation for onnxruntime-training (#15765) 2023-05-02 16:58:16 -07:00
optim support latest deepspeed version for optim (#15682) 2023-04-25 20:12:23 -07:00
ortmodule Python documentation for onnxruntime-training (#15765) 2023-05-02 16:58:16 -07:00
torchdynamo [DORT] Fix fake tensor problem cuased by PyTorch change (#15664) 2023-04-25 19:56:42 +08:00
utils Bump ruff in CI (#15533) 2023-04-17 10:11:44 -07:00
__init__.py Refining the offline tooling for training artifact generation (#15212) 2023-03-30 18:05:51 -07:00
_checkpoint_storage.py Enable pylint and numpy rules (#15218) 2023-03-27 20:37:53 -07:00
_utils.py Bump ruff in CI (#15533) 2023-04-17 10:11:44 -07:00
artifacts.py Python documentation for onnxruntime-training (#15765) 2023-05-02 16:58:16 -07:00
checkpoint.py Bump ruff in CI (#15533) 2023-04-17 10:11:44 -07:00
model_desc_validation.py
orttrainer.py
orttrainer_options.py
postprocess.py