check_if_semi_supervised

lightning_pose.utils.io.check_if_semi_supervised(losses_to_use: ListConfig | list | None = None) bool[source]

Use config file to determine if model is semi-supervised.

Take the entry of the hydra cfg that specifies losses_to_use. If it contains meaningful entries, infer that we want a semi_supervised model.

Parameters:

losses_to_use (ListConfig, list | None, optional) – the cfg entry specifying semisupervised losses to use. Defaults to None.

Returns:

True if the model is semi_supervised. False otherwise.

Return type:

bool