p2

stanfordnlp · Feb 4, 2025 · 913e9e7 · 913e9e7
1 parent ece9530
commit 913e9e7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/stanza/models/constituency/parser_training.py b/stanza/models/constituency/parser_training.py
@@ -627,7 +627,7 @@ def train_model_one_batch(epoch, batch_idx, model, training_batch, transition_te
             missing_node_errors.extend(Tree.single_missing_node_errors(gold_tree, reparsed_tree))
 
         if common_missing_nodes:
-            synthetic_trees = [x.tree.flip_missing_node_errors(common_missing_nodes) for x in training_batch]
+            synthetic_trees = [x.tree.flip_first_missing_node_error(common_missing_nodes) for x in training_batch]
             reparsed_results = model.analyze_trees(synthetic_trees, keep_constituents=True, keep_scores=False, keep_output_layers=True)
 
             reparsed_negatives = []