27 lines
720 B
Python
27 lines
720 B
Python
"""
|
|
Configuration utilities for loading and managing settings.
|
|
"""
|
|
import yaml
|
|
import os
|
|
from typing import Dict
|
|
|
|
|
|
def load_config(config_path: str = "config.yaml") -> Dict:
|
|
"""Load configuration from YAML file."""
|
|
if not os.path.exists(config_path):
|
|
raise FileNotFoundError(f"Config file not found: {config_path}")
|
|
|
|
with open(config_path, "r") as f:
|
|
config = yaml.safe_load(f)
|
|
|
|
return config
|
|
|
|
|
|
def create_directories(config: Dict):
|
|
"""Create necessary directories for checkpoints and logs."""
|
|
checkpoint_dir = config["training"]["checkpoint_dir"]
|
|
log_dir = config["training"]["log_dir"]
|
|
|
|
os.makedirs(checkpoint_dir, exist_ok=True)
|
|
os.makedirs(log_dir, exist_ok=True)
|