From 470059c65aa8d1c03e73789a9a602737844b097e Mon Sep 17 00:00:00 2001
From: piotrmp <piotr.m.przybyla@gmail.com>
Date: Tue, 13 Dec 2022 15:43:41 +0100
Subject: [PATCH] Bugfix.
---
src/lambo/learning/train.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/src/lambo/learning/train.py b/src/lambo/learning/train.py
index ce95d4d..fdbc805 100644
--- a/src/lambo/learning/train.py
+++ b/src/lambo/learning/train.py
@@ -168,7 +168,7 @@ def train_pretrained_and_save(language, treebank_path, save_path, pretrained_pat
print("Pretrained model not found, falling back to training from scratch.")
return train_new_and_save('LAMBO-BILSTM', treebank_path, save_path, epochs, device)
pretrained_model = torch.load(file_path, map_location=torch.device('cpu'))
- dict = Lambo.read_dict()
+ dict = Lambo.read_dict(pretrained_path / (pretrained_name + '.dict'))
print("Reading data.")
train_doc, dev_doc, test_doc = read_treebank(treebank_path, True)
--
GitLab