From 94c31516b20e1ed6dd3eb11d58afd0cf1cb3f8db Mon Sep 17 00:00:00 2001 From: John Bauer Date: Tue, 17 Dec 2024 18:57:32 -0800 Subject: [PATCH] Add flags for the contrastive learning to the model filename --- stanza/models/constituency_parser.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/stanza/models/constituency_parser.py b/stanza/models/constituency_parser.py index 47407024c5..a55576119b 100644 --- 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,