simplified model and configuration

2019-12-05 21:05:06 +01:00
parent 3a9a9f7861
commit a1994a71ee
3 changed files with 10 additions and 59 deletions
--- a/examples/summarization/run_summarization.py
+++ b/examples/summarization/run_summarization.py
@@ -31,9 +31,9 @@ Batch = namedtuple(

 def evaluate(args):
    tokenizer = BertTokenizer.from_pretrained("bert-base-uncased", do_lower_case=True)
-    model = bertabs = BertAbs.from_pretrained("bertabs-finetuned-cnndm")
-    bertabs.to(args.device)
-    bertabs.eval()
+    model = BertAbs.from_pretrained("bertabs-finetuned-cnndm")
+    model.to(args.device)
+    model.eval()

    symbols = {
        "BOS": tokenizer.vocab["[unused0]"],