| .. |
|
amp
|
Adopt linrtunner as the linting tool - take 2 (#15085)
|
2023-03-24 15:29:03 -07:00 |
|
api
|
Python documentation for onnxruntime-training (#15765)
|
2023-05-02 16:58:16 -07:00 |
|
experimental
|
Adopt linrtunner as the linting tool - take 2 (#15085)
|
2023-03-24 15:29:03 -07:00 |
|
onnxblock
|
Python documentation for onnxruntime-training (#15765)
|
2023-05-02 16:58:16 -07:00 |
|
optim
|
Adding this set_to_none flag to zero_grad to have signature parity with pytorch Adam (#16375)
|
2023-06-19 17:27:41 -07:00 |
|
ortmodule
|
Embedding sparsity optimization (#16141)
|
2023-06-19 20:34:53 +08:00 |
|
torchdynamo
|
DORT support for custom ops (#16392)
|
2023-06-21 15:58:05 -07:00 |
|
utils
|
Introduce memory observer for ORTModule (#16213)
|
2023-06-15 15:45:36 +08: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
|
Introduce float 8 types (#14731)
|
2023-05-30 13:25:58 -07:00 |
|
artifacts.py
|
Add ability to create ort format models from training offline utility (#16360)
|
2023-06-21 18:51:43 -07:00 |
|
checkpoint.py
|
Bump ruff in CI (#15533)
|
2023-04-17 10:11:44 -07:00 |
|
model_desc_validation.py
|
Adopt linrtunner as the linting tool - take 2 (#15085)
|
2023-03-24 15:29:03 -07:00 |
|
orttrainer.py
|
Adopt linrtunner as the linting tool - take 2 (#15085)
|
2023-03-24 15:29:03 -07:00 |
|
orttrainer_options.py
|
Adopt linrtunner as the linting tool - take 2 (#15085)
|
2023-03-24 15:29:03 -07:00 |
|
postprocess.py
|
Adopt linrtunner as the linting tool - take 2 (#15085)
|
2023-03-24 15:29:03 -07:00 |
