add parser parameter, add more parameters

supercoderhawk · Jan 10, 2020 · 8e8d613 · 8e8d613
1 parent df0376b
commit 8e8d613
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/wsdm_digg/reranking/parse_args.py b/wsdm_digg/reranking/parse_args.py
@@ -2,21 +2,28 @@
 import argparse
 
 
-def parse_args(args=None):
-    parser = argparse.ArgumentParser()
+def parse_args(args=None, parser=None):
+    if parser is None:
+        parser = argparse.ArgumentParser()
     parser.add_argument("-exp_name", required=False, type=str, help='')
     parser.add_argument("-train_filename", required=False, type=str, help='')
     parser.add_argument("-test_filename", required=False, type=str, help='')
+    parser.add_argument("-search_filename", required=False, type=str, help='')
     parser.add_argument("-dest_base_dir", required=False, type=str, help='')
     parser.add_argument("-batch_size", type=int, default=4, help='')
     parser.add_argument("-query_field", type=str, default='cites_text',
                         choices=['cites_text', 'description_text'], help='')
 
     parser.add_argument("-epoch", type=int, default=10, help='')
     parser.add_argument("-plm_learning_rate", type=float, default=1e-5, help='')
-    parser.add_argument("-ranker_learning_rate", type=float, default=1e-3, help='')
+    parser.add_argument("-rank_learning_rate", type=float, default=1e-3, help='')
+    parser.add_argument("-separate_learning_rate", action='store_true', help='')
     parser.add_argument("-save_model_step", type=int, default=2000, help='')
     parser.add_argument("-gradient_accumulate_step", type=int, default=4, help='')
+    parser.add_argument("-scheduler_lr", action='store_true', help='')
+    parser.add_argument("-scheduler_step", type=int, default=10000, help='')
+    parser.add_argument("-scheduler_gamma", type=float, default=0.5, help='')
+    parser.add_argument("-lazy_loading", action='store_true', help='')
 
     # model parameter
     parser.add_argument("-plm_model_name", required=False, type=str, help='')