From 7b6b1b503b1f71a3d92ee48023d651b6b4e9616f Mon Sep 17 00:00:00 2001
From: piotrmp <piotr.m.przybyla@gmail.com>
Date: Thu, 24 Nov 2022 09:03:02 +0100
Subject: [PATCH] Bug fix.
---
src/lambo/examples/run_training.py | 2 +-
src/lambo/examples/run_training_pretrained.py | 4 ++--
2 files changed, 3 insertions(+), 3 deletions(-)
diff --git a/src/lambo/examples/run_training.py b/src/lambo/examples/run_training.py
index 7bac54d..34c8b88 100644
--- a/src/lambo/examples/run_training.py
+++ b/src/lambo/examples/run_training.py
@@ -25,4 +25,4 @@ if __name__=='__main__':
continue
print(str(i) + '/' + str(len(languages)) + '========== ' + language + ' ==========')
inpath = treebanks / language
- train_new_and_save('LAMBO-BILSTM', inpath, outpath, device)
+ train_new_and_save('LAMBO-BILSTM', inpath, outpath, 10, device)
diff --git a/src/lambo/examples/run_training_pretrained.py b/src/lambo/examples/run_training_pretrained.py
index 33c3ea3..0e7f7ee 100644
--- a/src/lambo/examples/run_training_pretrained.py
+++ b/src/lambo/examples/run_training_pretrained.py
@@ -30,6 +30,6 @@ if __name__=='__main__':
print(str(i) + '/' + str(len(lines)) + '========== ' + model + ' ==========')
inpath = treebanks / model
if language != '?':
- train_pretrained_and_save(language, inpath, outpath, pretrained_path, device)
+ train_pretrained_and_save(language, inpath, outpath, pretrained_path, 10, device)
else:
- train_new_and_save('LAMBO-BILSTM', inpath, outpath, device)
+ train_new_and_save('LAMBO-BILSTM', inpath, outpath, 10, device)
--
GitLab