onnxruntime/orttraining/orttraining/python/training
Rui Ren db6a9bc033
support latest deepspeed version for optim (#15682)
### Description
<!-- Describe your changes. -->

support the latest deepspeed 0.9.1 for the next release


### Motivation and Context
<!-- - Why is this change required? What problem does it solve?
- If it fixes an open issue, please link to the issue here. -->
This will avoid the warn message `Skip modifying optimizer because of
unsupported DeepSpeed version`

---------

Co-authored-by: ruiren <ruiren@microsoft.com>
2023-04-25 20:12:23 -07:00
..
amp Adopt linrtunner as the linting tool - take 2 (#15085) 2023-03-24 15:29:03 -07:00
api C, C++, Python, C# API update for on device training (#15518) 2023-04-21 11:36:01 -07:00
experimental Adopt linrtunner as the linting tool - take 2 (#15085) 2023-03-24 15:29:03 -07:00
onnxblock C, C++, Python, C# API update for on device training (#15518) 2023-04-21 11:36:01 -07:00
optim support latest deepspeed version for optim (#15682) 2023-04-25 20:12:23 -07:00
ortmodule fix ORTModuleONNXModelException fallback OOM (#15523) 2023-04-25 15:10:31 -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 C, C++, Python, C# API update for on device training (#15518) 2023-04-21 11:36:01 -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