onnxruntime/orttraining/orttraining/python
pengwa 4f15b42728
Customize _get_tensor_rank for model export in stage3 (#18294)
### Customize _get_tensor_rank for model export in stage3

Weight/Params sizes are all (0), so exporter logic depending on input
shape will fail.

This PR override `_get_tensor_rank` function by retrieving the shape for
weight differently.


### 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. -->
2023-11-07 16:37:11 +08:00
..
deprecated Adopt linrtunner as the linting tool - take 2 (#15085) 2023-03-24 15:29:03 -07:00
training Customize _get_tensor_rank for model export in stage3 (#18294) 2023-11-07 16:37:11 +08:00
checkpointing_utils.py [Better Engineering] Bump ruff to 0.0.278 and fix new lint errors (#16789) 2023-07-21 12:53:41 -07:00
ort_trainer.py [Linter] Bump ruff and remove pylint (#17797) 2023-10-05 21:07:33 -07:00
orttraining_pybind_common.h
orttraining_pybind_state.cc Support inplace update for PythonOp/Grad (#17687) 2023-10-10 21:36:45 -07:00
orttraining_python_module.cc Python API to check whether collective ops are available or not (#17730) 2023-09-29 14:11:05 -07:00
orttraining_python_module_eager.h Run clang-format in CI (#15524) 2023-04-18 09:26:58 -07:00
pt_patch.py Adopt linrtunner as the linting tool - take 2 (#15085) 2023-03-24 15:29:03 -07:00