Temporarily disable @_weak_memoize

pyro-ppl · May 12, 2020 · 4986f82 · 4986f82
1 parent 89bb100
commit 4986f82
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyro/distributions/coalescent.py b/pyro/distributions/coalescent.py
@@ -229,7 +229,7 @@ def memoized_fn(*args):
 ))
 
 
-@_weak_memoize
+# @_weak_memoize  # DEBUG uncomment before committing
 @torch.no_grad()
 def _make_phylogeny(leaf_times, coal_times):
     assert leaf_times.size(-1) == 1 + coal_times.size(-1)