`torch_mist.estimators.discriminative.implementations.nwj`

Module Contents

Classes

class torch_mist.estimators.discriminative.implementations.nwj.NWJ(critic: torch_mist.critic.Critic, neg_samples: int = 1): Bases: torch_mist.estimators.discriminative.base.BaselineDiscriminativeMIEstimator