diff --git a/lib/cli/RLTraderCLI.py b/lib/cli/RLTraderCLI.py index d777fc1..c720f0f 100644 --- 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):