mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-25 22:26:24 +00:00
02333293de
### Description Motivation for this PR is code cleanup. 1. Remove all deprecated python code related to orttrainer, old checkpoint, related tests and utils 2. Cleanup orttraining_pybind_state.cc to remove all deprecated bindings.
31 lines
849 B
Python
31 lines
849 B
Python
# -------------------------------------------------------------------------
|
|
# Copyright (c) Microsoft Corporation. All rights reserved.
|
|
# Licensed under the MIT License.
|
|
# --------------------------------------------------------------------------
|
|
# isort: skip_file
|
|
from onnxruntime.capi._pybind_state import (
|
|
PropagateCastOpsStrategy,
|
|
TrainingParameters,
|
|
is_ortmodule_available,
|
|
)
|
|
|
|
# Options need to be imported before `ORTTrainer`.
|
|
from . import amp, artifacts, optim
|
|
|
|
__all__ = [
|
|
"PropagateCastOpsStrategy",
|
|
"TrainingParameters",
|
|
"is_ortmodule_available",
|
|
"amp",
|
|
"artifacts",
|
|
"optim",
|
|
]
|
|
|
|
try:
|
|
if is_ortmodule_available():
|
|
from .ortmodule import ORTModule # noqa: F401
|
|
|
|
__all__.append("ORTModule")
|
|
except ImportError:
|
|
# That is OK iff this is not a ORTModule training package
|
|
pass
|