From 9a7523c4579b0069b05bcbd4d8e5dd3b15128b14 Mon Sep 17 00:00:00 2001
From: piotrmp <piotr.m.przybyla@gmail.com>
Date: Wed, 23 Nov 2022 10:01:46 +0100
Subject: [PATCH] Bug fix.
---
src/lambo/examples/run_pretraining.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/src/lambo/examples/run_pretraining.py b/src/lambo/examples/run_pretraining.py
index 45531bc..a6d0173 100644
--- a/src/lambo/examples/run_pretraining.py
+++ b/src/lambo/examples/run_pretraining.py
@@ -16,8 +16,8 @@ from lambo.learning.train import pretrain
from lambo.utils.oscar import read_jsonl_to_documents, download_archive1_from_oscar
if __name__=='__main__':
- outpath = sys.argv[1] #Path.home() / 'PATH-TO/models/pretrained/'
- tmppath = sys.argv[2] #Path.home() / 'PATH-TO/tmp/tmp.jsonl.gz'
+ outpath = Path(sys.argv[1]) #Path.home() / 'PATH-TO/models/pretrained/'
+ tmppath = Path(sys.argv[2]) #Path.home() / 'PATH-TO/tmp/tmp.jsonl.gz'
# These need to be filled ine before running. OSCAR is avaialable on request.
OSCAR_LOGIN = sys.argv[3]
OSCAR_PASSWORD = sys.argv[4]
--
GitLab