Spaces:
Sleeping
Sleeping
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"
)
|