1
Fork 0

feat: allow model metadata cli override

This commit is contained in:
Tibo De Peuter 2026-04-30 20:19:42 +02:00
parent ba8184b034
commit be78fb15bd
Signed by: tdpeuter
SSH key fingerprint: SHA256:u/h/LVoqKF1Iz02uOyxe6hcjmoZASCGV2HM0TG9ZMoU
5 changed files with 58 additions and 7 deletions

View file

@ -51,7 +51,11 @@ def main(dict_cfg: DictConfig) -> None:
raise ValueError(f"Expected a '.flax' checkpoint, got '{model_path.name}'.")
# 2. Discover + load sidecar metadata YAML
metadata = load_metadata(model_path)
metadata_override = None
if sim_cfg.metadata_path is not None:
metadata_override = Path(hydra.utils.to_absolute_path(sim_cfg.metadata_path))
metadata = load_metadata(model_path, metadata_override)
# 3. Reconstruct typed configs from metadata
training = metadata_to_configs(metadata)