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
..
amp	[Better Engineering] Bump ruff to 0.0.278 and fix new lint errors (#16789)	2023-07-21 12:53:41 -07:00
api	[On-Device Training] Expose Parameters through the Training API (#17364)	2023-09-25 20:03:24 -07:00
experimental	Manage ORTModule configurations consistently (#16396)	2023-06-27 19:19:36 +08:00
onnxblock	[Linter] Bump ruff and remove pylint (#17797)	2023-10-05 21:07:33 -07:00
optim	FP16 optimizer automatically detect DeepSpeed compatibility (#18084)	2023-10-25 15:11:02 +08:00
ort_triton	Fix PoliCheck (#18180)	2023-10-31 13:53:11 +08:00
ortmodule	Customize _get_tensor_rank for model export in stage3 (#18294)	2023-11-07 16:37:11 +08:00
torchdynamo	Fix orttraining_test_dort.py (#17034)	2023-08-08 08:11:48 -07:00
utils	Optimize 4bit Qlora training (#18131)	2023-11-02 09:46:11 -07:00
__init__.py	Add mac and windows python packages for onnxruntime-training (#16993)	2023-08-07 20:32:55 -07:00
_checkpoint_storage.py	Enable pylint and numpy rules (#15218)	2023-03-27 20:37:53 -07:00
_utils.py	[Better Engineering] Bump ruff to 0.0.278 and fix new lint errors (#16789)	2023-07-21 12:53:41 -07:00
artifacts.py	add logits option to generate artifacts (#17276)	2023-08-29 16:55:31 -07:00
checkpoint.py	Disable PERF* rules in ruff to allow better readability (#16834)	2023-07-25 15:38:22 -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	[Linter] Bump ruff and remove pylint (#17797)	2023-10-05 21:07:33 -07:00
orttrainer_options.py	[Better Engineering] Bump ruff to 0.0.278 and fix new lint errors (#16789)	2023-07-21 12:53:41 -07:00
postprocess.py	Disable PERF* rules in ruff to allow better readability (#16834)	2023-07-25 15:38:22 -07:00