Add flags for the contrastive learning to the model filename

stanfordnlp · Dec 18, 2024 · 867d8da · 867d8da
1 parent 4bce015
commit 867d8da
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/stanza/models/constituency_parser.py b/stanza/models/constituency_parser.py
@@ -727,6 +727,8 @@ def build_model_filename(args):
                                                oracle_level=args['oracle_level'],
                                                embedding=embedding,
                                                finetune=maybe_finetune,
+                                               contrastive_learning_rate=args['contrastive_learning_rate'],
+                                               contrastive_initial_epoch=args['contrastive_initial_epoch'],
                                                transformer_finetune_begin=transformer_finetune_begin,
                                                transition_scheme=args['transition_scheme'].name.lower().replace("_", ""),
                                                tscheme=args['transition_scheme'].short_name,