ckpt_path_from_base_path

lightning_pose.utils.io.ckpt_path_from_base_path(base_path: str, model_name: str, logging_dir_name: str = 'tb_logs/') → str | None[source]

Given a path to a hydra output with trained model, extract the model .ckpt file.

Parameters:

  • base_path (str) – path to a folder with logs and checkpoint. for example, function will search base_path/logging_dir_name/model_name…

  • model_name (str) – the name you gave your model before training it; appears as model_name in lightning-pose/scripts/config/model_params.yaml

  • logging_dir_name (str, optional) – name of the folder in logs, controlled in train_hydra.py Defaults to “tb_logs/”.

  • version (int. optional)

Returns:

path to model checkpoint, or None if none found.

Return type:

str