option --epochs enabled

yzhangcs · yzhangcs · commit 209de68b9502 · 2020-12-23T20:16:13.000+09:00
diff --git a/supar/cmds/cmd.py b/supar/cmds/cmd.py
@@ -13,7 +13,8 @@ def parse(parser):
     parser.add_argument('--seed', '-s', default=1, type=int, help='seed for generating random numbers')
     parser.add_argument('--threads', '-t', default=16, type=int, help='max num of threads')
     parser.add_argument('--batch-size', default=5000, type=int, help='batch size')
-    parser.add_argument("--local_rank", type=int, default=-1, help='node rank for distributed training')
+    parser.add_argument('--local_rank', default=-1, type=int, help='node rank for distributed training')
+    parser.add_argument('--epochs', default=5000, type=int, help='epochs')
     args, unknown = parser.parse_known_args()
     args, _ = parser.parse_known_args(unknown, args)
     args = Config(**vars(args))
