`torch_mist.estimators.discriminative.implementations.mine`

Module Contents

Classes

class torch_mist.estimators.discriminative.implementations.mine.MINE(critic: torch_mist.critic.Critic, neg_samples: int = 1, gamma: float = 0.9)

Bases: torch_mist.estimators.discriminative.base.BaselineDiscriminativeMIEstimator

lower_bound = False

train_baseline()

batch_loss(x: torch.Tensor, y: torch.Tensor) → torch.Tensor