onnxruntime/orttraining/orttraining/python/training
Ti-Tai Wang 45ea907f53
Fix orttraining_test_dort.py (#17034)
Converter has moved `opset_version` out from `torch.onnx.ExportOptions`,
and put it into `torch.onnx.OnnxRegistry`.
This PR fixes the usage in DORT.
2023-08-08 08:11:48 -07: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 Add support for training apis to support custom ops (#16601) 2023-07-14 11:15:51 -07:00
experimental Manage ORTModule configurations consistently (#16396) 2023-06-27 19:19:36 +08:00
onnxblock Disable PERF* rules in ruff to allow better readability (#16834) 2023-07-25 15:38:22 -07:00
optim [Better Engineering] Bump ruff to 0.0.278 and fix new lint errors (#16789) 2023-07-21 12:53:41 -07:00
ort_triton Disable PERF* rules in ruff to allow better readability (#16834) 2023-07-25 15:38:22 -07:00
ortmodule Add mac and windows python packages for onnxruntime-training (#16993) 2023-08-07 20:32:55 -07:00
torchdynamo Fix orttraining_test_dort.py (#17034) 2023-08-08 08:11:48 -07:00
utils Refactor schema extraction and output unflattening (#16894) 2023-08-04 13:58:21 +08: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 support for training apis to support custom ops (#16601) 2023-07-14 11:15:51 -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 Disable PERF* rules in ruff to allow better readability (#16834) 2023-07-25 15:38:22 -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