From 671f6e056dd2a3f51ac5843f8810f5ff68d5c85b Mon Sep 17 00:00:00 2001 From: Iz Beltagy Date: Tue, 5 May 2020 08:16:44 -0700 Subject: [PATCH] Update triviaqa.py --- scripts/triviaqa.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/scripts/triviaqa.py b/scripts/triviaqa.py index 5602f73..6277088 100644 --- a/scripts/triviaqa.py +++ b/scripts/triviaqa.py @@ -265,7 +265,7 @@ def __init__(self, args): self.train_dataloader_object = self.val_dataloader_object = self.test_dataloader_object = None def load_model(self): - model = Longformer.from_pretrained(args.model_path) + model = Longformer.from_pretrained(self.args.model_path) for layer in model.encoder.layer: layer.attention.self.attention_mode = self.args.attention_mode self.args.attention_window = layer.attention.self.attention_window