get_model

lightning_pose.utils.scripts.get_model(cfg: DictConfig, data_module: BaseDataModule | UnlabeledDataModule | None, loss_factories: dict[str, LossFactory] | dict[str, None]) → LightningModule

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