`torch_mist.estimators.discriminative.implementations.alpha_tuba`

Module Contents

Classes

class torch_mist.estimators.discriminative.implementations.alpha_tuba.AlphaTUBA(critic: torch_mist.critic.Critic, baseline: torch_mist.baseline.LearnableBaseline, alpha: float = 0.01, neg_samples: int = -1): Bases: torch_mist.estimators.discriminative.base.BaselineDiscriminativeMIEstimator