Skip to content
Permalink
Branch: master
Find file Copy path
Find file Copy path
Fetching contributors…
Cannot retrieve contributors at this time
33 lines (24 sloc) 845 Bytes
"""
Description:
This script builds the model and trains it using user-specified input data.
Output:
Saves model to a user-specified output file.
Usage:
selene_cli.py <config-yml> [--lr=<lr>]
selene_cli.py -h | --help
Options:
-h --help Show this screen.
<config-yml> Model-specific parameters
--lr=<lr> If training, the optimizer's learning rate
[default: None]
"""
from docopt import docopt
from selene_sdk.utils import load_path
from selene_sdk.utils import parse_configs_and_run
from selene_sdk import __version__
if __name__ == "__main__":
arguments = docopt(
__doc__,
version=__version__)
configs = load_path(arguments["<config-yml>"], instantiate=False)
parse_configs_and_run(configs, lr=arguments["--lr"])
You can’t perform that action at this time.