get_model

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

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