get_loss_factories

lightning_pose.losses.factory.get_loss_factories(cfg: DictConfig | ListConfig, data_module: BaseDataModule | UnlabeledDataModule) → dict[source]

Create supervised and unsupervised loss factories from a hydra config.

Parameters:

  • cfg – hydra config carrying model, data, and loss parameters.

  • data_module – data module passed to data-dependent losses such as PCA.

Returns:

dict with keys 'supervised' and 'unsupervised', each mapping to a LossFactory instance.