Replace --proc-number with --jobs

notadamking · Jul 6, 2019 · ac3820a · ac3820a
1 parent 858ebef
commit ac3820a
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/lib/cli/RLTraderCLI.py b/lib/cli/RLTraderCLI.py
@@ -39,6 +39,8 @@ def __init__(self):
             help='Tensorboard path',
             dest='tensorboard_path'
         )
+        self.parser.add_argument('--proc-number', type=int, default=multiprocessing.cpu_count(),
+                                 help='How many processes to spawn')
 
         subparsers = self.parser.add_subparsers(help='Command', dest="command")
 
@@ -48,14 +50,9 @@ def __init__(self):
         opt_train_test_parser.add_argument('--train-epochs', type=int, default=10, help='Train for how many epochs')
         opt_train_test_parser.add_argument('--no-render', action='store_false', help='Should render the model')
         opt_train_test_parser.add_argument('--no-test', action='store_false', help='Should test the model')
-        opt_train_test_parser.add_argument('--proc-number', type=int, default=multiprocessing.cpu_count(),
-                                           help='How many processes to spawn')
 
         optimize_parser = subparsers.add_parser('optimize', description='Optimize model parameters')
         optimize_parser.add_argument('--trials', type=int, default=1, help='Number of trials')
-        optimize_parser.add_argument('--parallel-jobs', type=int, default=1, help='How many jobs in parallel')
-        optimize_parser.add_argument('--proc-number', type=int, default=multiprocessing.cpu_count(),
-                                     help='How many processes to spawn')
 
         optimize_parser.add_argument('--verbose-model', type=int, default=1, help='Verbose model', dest='model_verbose')
 
@@ -65,6 +62,9 @@ def __init__(self):
         test_parser = subparsers.add_parser('test', description='Test model')
         test_parser.add_argument('--model-epoch', type=int, default=1, help='Model epoch index')
         test_parser.add_argument('--no-render', action='store_false', help='Do not render test')
+
+        subparsers.add_parser('update-static', description='Update static data')
+
         self.parser.set_defaults(**defaults)
 
     def get_args(self):