11 lines
298 B
Python
Executable File
11 lines
298 B
Python
Executable File
from ..config import cfg
|
|
|
|
from ..utils.distributed import fix_unset_envs
|
|
fix_unset_envs()
|
|
|
|
if cfg.trainer.backend == "deepspeed":
|
|
from .deepspeed import Engine
|
|
elif cfg.trainer.backend == "local":
|
|
from .base import Engine
|
|
|
|
from .base import Engines, TrainFeeder, default_feeder, Engine as _Engine |