Final polish

This commit is contained in:
Johnny Fernandes
2026-05-14 21:16:03 +01:00
parent 3bff7eefb0
commit afd26f47d2
732 changed files with 4149 additions and 79134 deletions
+2 -2
View File
@@ -25,8 +25,8 @@ PHASE5_RUNS = {
def _load_cfg(name: str) -> dict:
with open(CFG_DIR / PHASE5_RUNS[name]["config"]) as f:
return json.load(f)
from src.utils import load_config
return load_config(str(CFG_DIR / PHASE5_RUNS[name]["config"]))
def _load_model(name: str, cfg: dict, device: torch.device):