Language Language English 简体中文

unilab.training.run.parse_checkpoint_path

unilab.training.run.parse_checkpoint_path(cfg, *, root_dir, load_run=None, task_name=None, checkpoint=None, suffix='.pt')

Resolve a checkpoint path from Hydra config and repository root.

Parameters:

• cfg (DictConfig)

• root_dir (str | Path)

• load_run (str | int | PathLike[str] | None)

• task_name (str | None)

• checkpoint (str | int | None)

• suffix (str)

Return type:

tuple[Path | None, Path | None]