diff --git a/src/lambo/examples/run_training.py b/src/lambo/examples/run_training.py
index 95544040b5109af203e2e05b3786f7459487f935..5676f365d2049fce86349966c1882fd59d3c6f6e 100644
--- a/src/lambo/examples/run_training.py
+++ b/src/lambo/examples/run_training.py
@@ -27,4 +27,4 @@ if __name__=='__main__':
             continue
         print(str(i) + '/' + str(len(languages)) + '========== ' + language + ' ==========')
         inpath = treebanks / language
-        train_new_and_save('LAMBO-BILSTM', inpath, outpath, 10, device)
+        train_new_and_save('LAMBO-BILSTM', inpath, outpath, 20, device)
diff --git a/src/lambo/examples/run_training_pretrained.py b/src/lambo/examples/run_training_pretrained.py
index 6a81eb240a76e4e4389af7e36634f583b894c12c..9cc6b2bb0be8df0aafe09032d4b503458d5b2bc3 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, 10, device)
+            train_pretrained_and_save(language, inpath, outpath, pretrained_path, 20, device)
         else:
-            train_new_and_save('LAMBO-BILSTM', inpath, outpath, 10, device)
+            train_new_and_save('LAMBO-BILSTM', inpath, outpath, 20, device)