/MIGE

Mutual Information Gradient Estimation

Primary LanguageJupyter NotebookMIT LicenseMIT

Mutual Information Gradient Estimation

Requirements:

  • pytorch >= 1.2.0
  • numpy
  • tqdm
  • matplotlib