Bug fix.

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)

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)