()
| 9 | |
| 10 | |
| 11 | def run_task(): |
| 12 | assert hparams['task_cls'] != '' |
| 13 | pkg = ".".join(hparams["task_cls"].split(".")[:-1]) |
| 14 | cls_name = hparams["task_cls"].split(".")[-1] |
| 15 | task_cls = getattr(importlib.import_module(pkg), cls_name) |
| 16 | task_cls.start() |
| 17 | |
| 18 | def clear_gpus(): |
| 19 | devices = os.environ.get('CUDA_VISIBLE_DEVICES', '').split(",") |