Update VerbAtlas version (1.1)

Riccorl · Sep 24, 2020 · e61bcb6 · e61bcb6
1 parent 4891be3
commit e61bcb6
Show file tree

Hide file tree

Showing 5 changed files with 5,779 additions and 5,826 deletions.
diff --git a/setup.py b/setup.py
@@ -5,7 +5,7 @@
 
 setuptools.setup(
     name="transformer_srl",  # Replace with your own username
-    version="2.4rc2",
+    version="2.4rc3",
     author="Riccardo Orlando",
     author_email="orlandoricc@gmail.com",
     description="SRL Transformer model",

diff --git a/transformer_srl/models.py b/transformer_srl/models.py
@@ -337,7 +337,7 @@ def __init__(
         self.frame_criterion = torch.nn.CrossEntropyLoss()
         # add missing labels
         frame_list = load_label_list(FRAME_LIST_PATH)
-        # self.vocab.add_tokens_to_namespace(frame_list, "frames_labels")
+        self.vocab.add_tokens_to_namespace(frame_list, "frames_labels")
         self.num_classes = self.vocab.get_vocab_size("labels")
         self.frame_num_classes = self.vocab.get_vocab_size("frames_labels")
         if srl_eval_path is not None: