yetrun's picture
ver3: 将源码迁入 src/deep_learning 包,重塑训练流水线,规范 data/model 契约
07cb7d3
Raw
History Blame Contribute Delete
757 Bytes
"""
流水线路径工具。
负责把任务名和可选 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"
)