Update tactic_decoder.py

princeton-vl · May 8, 2021 · 3dbb43e · 3dbb43e
1 parent e4e5f10
commit 3dbb43e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ASTactic/models/tactic_decoder.py b/ASTactic/models/tactic_decoder.py
@@ -607,7 +607,7 @@ def beam_search(self, environment, local_context, goal):
                     node.expand(action)
                 new_beam.append(ast)
                 new_frontiers.append(stack)
-                new_log_likelihood.append(log_cond_prob)
+                new_log_likelihood.append(log_likelihood[idx] + log_cond_prob)
             beam = new_beam
             frontiers = new_frontiers
             log_likelihood = new_log_likelihood