| """Application-wide configuration and resource paths.""" |
|
|
| import torch |
| import yaml |
| import os |
| from pathlib import Path |
| from types import SimpleNamespace |
|
|
| project_root = Path(__file__).resolve().parent |
|
|
| device = torch.device("cuda" if torch.cuda.is_available() else "cpu") |
|
|
| vae_model_path = str(project_root / "models" / "vae.pth") |
| unet_model_path = str(project_root / "models" / "unet.pth") |
| ema_unet_model_path = str(project_root / "models" / "ema-unet.pth") |
|
|
| vae_plots_path = str(project_root / "plots" / "vae") |
| unet_plots_path = str(project_root / "plots" / "unet") |
|
|
| |
| config_path = os.path.join(project_root, "config.yaml") |
| with open(config_path, "r") as f: |
| cfg_dict = yaml.safe_load(f) |
|
|
| cfg = SimpleNamespace(**cfg_dict) |
|
|
