Skip to content
Snippets Groups Projects
Commit 7b6b1b50 authored by piotrmp's avatar piotrmp
Browse files

Bug fix.

parent 88413d8f
Branches
1 merge request!1Migration to UD 2.11
Hide whitespace changes
Inline Side-by-side
src/lambo/examples/run_training.py
+ 1
− 1
View file @ 7b6b1b50
...@@ -25,4 +25,4 @@ if __name__=='__main__': ...@@ -25,4 +25,4 @@ if __name__=='__main__':
continue continue
print(str(i) + '/' + str(len(languages)) + '========== ' + language + ' ==========') print(str(i) + '/' + str(len(languages)) + '========== ' + language + ' ==========')
inpath = treebanks / language inpath = treebanks / language
train_new_and_save('LAMBO-BILSTM', inpath, outpath, device) train_new_and_save('LAMBO-BILSTM', inpath, outpath, 10, device)
src/lambo/examples/run_training_pretrained.py
+ 2
− 2
View file @ 7b6b1b50
...@@ -30,6 +30,6 @@ if __name__=='__main__': ...@@ -30,6 +30,6 @@ if __name__=='__main__':
print(str(i) + '/' + str(len(lines)) + '========== ' + model + ' ==========') print(str(i) + '/' + str(len(lines)) + '========== ' + model + ' ==========')
inpath = treebanks / model inpath = treebanks / model
if language != '?': if language != '?':
train_pretrained_and_save(language, inpath, outpath, pretrained_path, device) train_pretrained_and_save(language, inpath, outpath, pretrained_path, 10, device)
else: else:
train_new_and_save('LAMBO-BILSTM', inpath, outpath, device) train_new_and_save('LAMBO-BILSTM', inpath, outpath, 10, device)
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment