12 lines
163 B
Python
Executable File
12 lines
163 B
Python
Executable File
__all__ = [
|
|
'build_trainer'
|
|
]
|
|
|
|
from utils import Registry
|
|
|
|
TRAINER = Registry()
|
|
|
|
def build_trainer(cfg):
|
|
trainer = TRAINER[cfg["task"]]
|
|
return trainer
|