minor fix

thu-ml · Apr 8, 2019 · 62584ab · 62584ab
1 parent d738c27
commit 62584ab
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/topic_models/lntm_mcem.py b/examples/topic_models/lntm_mcem.py
@@ -68,7 +68,7 @@ def lntm(n_chains, n_docs, n_topics, n_vocab, eta_mean, eta_logstd):
     num_e_steps = 5
     hmc = zs.HMC(step_size=1e-3, n_leapfrogs=20, adapt_step_size=True,
                  target_acceptance_rate=0.6)
-    epochs = 1
+    epochs = 100
     learning_rate_0 = 1.0
     t0 = 10