diff --git a/ASTactic/models/tactic_decoder.py b/ASTactic/models/tactic_decoder.py index fa171282..11d3381e 100644 --- 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