saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-24 19:43:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
onnxruntime/orttraining/orttraining/python/training
History
pengwa 34fe8fb069
Type hint for ORTModule (#15938) 
### Type hint for ORTModule

Add Type hint for ORTModule
Refine comments. 

The reason of removing theinterface execution_session_run_forward from
`orttraining/orttraining/python/training/ortmodule/_graph_execution_manager.py`:

PR
cc275e7529 (diff-497e18dc8878818205b81fd80f85942548d8aa15d0f1204ce3e3d9795e3dd195)
and some commit before it breaks the function interface contracts
between parent calss _graph_execution_manager.py and its children
_training_manager.py and _inference_manager.py. So there is no need to
have this interface.


### Other EE work opportunities

1. Use logger correctly. 
2. Remove few duplication logic parsing input/output recursively.
3. Clean up environment variable usage.
2023-05-25 09:28:20 +08:00
..
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 support latest deepspeed version for optim (#15682) 2023-04-25 20:12:23 -07:00
ortmodule Type hint for ORTModule (#15938) 2023-05-25 09:28:20 +08: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 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