change max_pos to 512

nlpyang · Sep 2, 2019 · 0e0905d · 0e0905d
1 parent 752cb19
commit 0e0905d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/train.py b/src/train.py
@@ -38,7 +38,7 @@ def str2bool(v):
     parser.add_argument("-batch_size", default=140, type=int)
     parser.add_argument("-test_batch_size", default=200, type=int)
 
-    parser.add_argument("-max_pos", default=800, type=int)
+    parser.add_argument("-max_pos", default=512, type=int)
     parser.add_argument("-use_interval", type=str2bool, nargs='?',const=True,default=True)
     parser.add_argument("-large", type=str2bool, nargs='?',const=True,default=False)
     parser.add_argument("-load_from_extractive", default='', type=str)