mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-28 22:56:32 +00:00
19 lines
883 B
Python
19 lines
883 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 # noqa: F401
|
|
from onnxruntime.capi.training.training_session import TrainingSession # noqa: F401
|
|
|
|
|
|
# Options need to be imported before `ORTTrainer`.
|
|
from .orttrainer_options import ORTTrainerOptions # noqa: F401
|
|
from .orttrainer import ORTTrainer, TrainStepInfo # noqa: F401
|
|
from . import amp, artifacts, checkpoint, model_desc_validation, optim # noqa: F401
|
|
|
|
try: # noqa: SIM105
|
|
from .ortmodule import ORTModule # noqa: F401
|
|
except ImportError:
|
|
# That is OK iff this is not a ORTModule training package
|
|
pass
|