(ckpt_path: Path)
| 77 | |
| 78 | |
| 79 | def _find_run_dir_for_checkpoint(ckpt_path: Path) -> Path: |
| 80 | # expected: outputs/hora/revo3_right/<run_name>/stage{1,2}_nn/*.pth|*.ckpt |
| 81 | if ckpt_path.parent.name.endswith("_nn"): |
| 82 | return ckpt_path.parent.parent |
| 83 | return ckpt_path.parent |
| 84 | |
| 85 | |
| 86 | def _load_config(path: Path | None) -> Any | None: |