Created using Colaboratory

Author: lollcat

experiments/gmm/fab_gmm.ipynb

@@ -235,7 +235,9 @@
         "alpha = 2.0\n",
         "\n",
         "# AIS\n",
-        "transition_operator_type = \"metropolis\"\n",
+        "# By default we use a simple metropolis mcmc transition with a fixed step size.\n",
+        "# Can switch this to 'hmc' to improve training efficiency. \n",
+        "transition_operator_type = \"metropolis\" \n",
         "n_intermediate_distributions = 1\n",
         "metropolis_step_size = 5.0\n",
         "\n",