torch_mist.utils.evaluation.mi_estimator

Module Contents

Functions

evaluate_mi(, batch_size, num_workers, Dict[str, float]])

torch_mist.utils.evaluation.mi_estimator.evaluate_mi(estimator: torch_mist.estimators.base.MIEstimator, data: torch_mist.utils.data.utils.TensorDictLike, device: torch.device = torch.device('cpu'), batch_size: int | None = None, num_workers: int = 0) → float | Dict[str, float]