Final polish
This commit is contained in:
@@ -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):
|
||||
|
||||
Reference in New Issue
Block a user