train

lightning_pose.train.train(cfg: DictConfig | ListConfig, model_dir: str | Path | None = None, skip_evaluation: bool = False) → Model

Train a model using the configuration cfg, saving outputs to model_dir.

Parameters:

• cfg – hydra config object.

• model_dir – directory to save model outputs; defaults to cwd if unspecified.

• skip_evaluation – if True, skip post-training evaluation.

Returns:

trained Model instance.