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