File size: 757 Bytes
14f6839
 
 
 
 
 
 
 
 
07cb7d3
 
14f6839
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
"""
流水线路径工具。

负责把任务名和可选 task_dir 解析成 PipelineRuntime,统一确定日志目录、检查点目录和
tensorboard 目录。
"""

from pathlib import Path

from deep_learning.env.resolve import resolve_task_dir
from deep_learning.pipeline.context import PipelineRuntime


def build_runtime(name: str, task_dir: Path | None = None) -> PipelineRuntime:
    resolved_task_dir = task_dir
    if resolved_task_dir is None:
        resolved_task_dir = resolve_task_dir(name)

    return PipelineRuntime(
        name=name,
        task_dir=resolved_task_dir,
        log_dir=resolved_task_dir / "logs",
        checkpoint_dir=resolved_task_dir / "checkpoints",
        tensorboard_dir=resolved_task_dir / "tensorboard"
    )