File size: 1,470 Bytes
86fe6bc | 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 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 | """
配置管理模块 — 公共模块 (Person E 可以先完成)
功能: 加载 YAML 配置，支持命令行覆盖和多配置合并。
"""
from __future__ import annotations
import logging
from pathlib import Path
from typing import Any, Optional
from omegaconf import OmegaConf, DictConfig
logger = logging.getLogger(__name__)
def load_config(config_path: str | Path) -> DictConfig:
"""
加载 YAML 配置文件。
TODO [Person E]: 实现以下逻辑:
1. 使用 OmegaConf.load(config_path) 加载配置
2. 验证配置完整性
3. 返回 DictConfig 对象
"""
raise NotImplementedError("TODO: Person E 实现 load_config")
def merge_configs(base_config: DictConfig, override_config: DictConfig) -> DictConfig:
"""
合并配置 (override 覆盖 base)。
TODO [Person E]:
使用 OmegaConf.merge(base_config, override_config)
"""
raise NotImplementedError("TODO: Person E 实现 merge_configs")
def config_from_cli(config_path: str, cli_args: list[str]) -> DictConfig:
"""
从配置文件 + 命令行参数构建最终配置。
TODO [Person E]:
1. 加载 config_path
2. 使用 OmegaConf.from_cli(cli_args) 解析命令行参数
3. 合并并返回
使用方式: python train.py --config configs/default.yaml training.lr=1e-4 model.nhead=16
"""
raise NotImplementedError("TODO: Person E 实现 config_from_cli")
|