18 lines
388 B
Python
18 lines
388 B
Python
|
import importlib
|
||
|
|
||
|
from utils.hparams import set_hparams, hparams
|
||
|
|
||
|
set_hparams(print_hparams=False)
|
||
|
|
||
|
|
||
|
def run_task():
|
||
|
assert hparams['task_cls'] != ''
|
||
|
pkg = ".".join(hparams["task_cls"].split(".")[:-1])
|
||
|
cls_name = hparams["task_cls"].split(".")[-1]
|
||
|
task_cls = getattr(importlib.import_module(pkg), cls_name)
|
||
|
task_cls.start()
|
||
|
|
||
|
|
||
|
if __name__ == '__main__':
|
||
|
run_task()
|