diff --git a/src/lambo/examples/run_training.py b/src/lambo/examples/run_training.py
index 7bac54dee19906cbad72e25b79acb3b6b710edcb..34c8b889b55829764a7d0710172e31849e7b3035 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 33c3ea3078a965a7d03ac99b40e5e5c5df1a775c..0e7f7ee88cdbf2851aeab3a6ff7579efa2666666 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)