Пример #1
0
def load_torch_model_training_from_task_id(model_cls: Type[TorchModelTraining],
                                           task_id: str) -> TorchModelTraining:
    task_dir = get_task_dir(model_cls, task_id)
    if not os.path.exists(task_dir):
        task_dir = get_interaction_dir(model_cls, task_id)

    return load_torch_model_training_from_task_dir(model_cls, task_dir)
Пример #2
0
 def output(self):
     return luigi.LocalTarget(get_task_dir(self.__class__, self.task_id))