add --epochs to hyperopt subcommand

This commit is contained in:
gcarq
2017-11-25 01:12:44 +01:00
parent b9c4eafd96
commit 9ff1f05e66
4 changed files with 34 additions and 8 deletions

View File

@@ -159,7 +159,14 @@ def build_subcommands(parser: argparse.ArgumentParser) -> None:
# Add hyperopt subcommand
hyperopt_cmd = subparsers.add_parser('hyperopt', help='hyperopt module')
hyperopt_cmd.set_defaults(func=hyperopt.start)
hyperopt_cmd.add_argument(
'-e', '--epochs',
help='specify number of epochs (default: 100)',
dest='epochs',
default=100,
type=int,
metavar='INT',
)
# Required json-schema for user specified config
CONF_SCHEMA = {