2021/9/15 nezha_span

isLouisHsu · Sep 15, 2021 · f442600 · f442600
1 parent cadbc56
commit f442600
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/run_span.py b/run_span.py
@@ -1107,7 +1107,7 @@ def predict(args, model, processor, tokenizer, prefix=""):
 
 MODEL_CLASSES = {
     "bert_span": (BertConfigSpanV2, BertSpanV2ForNer, BertTokenizer),
-    # "nezha_span": (BertConfigSpanV2, NeZhaSpanV2ForNer, BertTokenizer),
+    "nezha_span": (BertConfigSpanV2, NeZhaSpanV2ForNer, BertTokenizer),
 }
 
 def load_dataset(args, processor, tokenizer, data_type='train'):