hyrax.verbs.train

Attributes

logger

Classes

Train

Train verb

Module Contents

logger[source]
class Train(config)[source]

Bases: hyrax.verbs.verb_registry.Verb

Train verb

__init__()[source]

Overall initialization for all verbs that saves the config

cli_name = 'train'[source]
add_parser_kwargs[source]
static setup_parser(parser)[source]

We don’t need any parser setup for CLI opts

run_cli(args=None)[source]

CLI stub for Train verb

run()[source]

Run the training process for the configured model and data loader. Returns the trained model.

static _log_params(config, results_dir)[source]

Log the various parameters to mlflow from the config file.

Parameters:
  • config (dict) – The main configuration dictionary

  • results_dir (str) – The full path to the results sub-directory