get_model

lightning_pose.models.get_model(cfg: DictConfig | ListConfig, data_module: BaseDataModule | UnlabeledDataModule | None, loss_factories: dict[str, LossFactory] | dict[str, None]) → ALLOWED_MODELS

Create model: regression or heatmap based, supervised or semi-supervised.