cli.utils.train
cli.utils.train
Utilities for axolotl train CLI command.
Functions
Name | Description |
---|---|
build_command | Build command list from base command and options. |
generate_config_files | Generate list of configuration files to process. Yields a tuple of the configuration file name and a boolean indicating |
launch_training | Execute training with the given configuration. |
build_command
cli.utils.train.build_command(base_cmd, options)
Build command list from base command and options.
Parameters
Name | Type | Description | Default |
---|---|---|---|
base_cmd | list[str] | Command without options. | required |
options | dict[str, Any] | Options to parse and append to base command. | required |
Returns
Name | Type | Description |
---|---|---|
list[str] | List of strings giving shell command. |
generate_config_files
cli.utils.train.generate_config_files(config, sweep)
Generate list of configuration files to process. Yields a tuple of the configuration file name and a boolean indicating whether this is a group of configurations (i.e., a sweep).
Parameters
Name | Type | Description | Default |
---|---|---|---|
config | str | Base configuration file | required |
sweep | str | None | Sweep configuration file | required |
launch_training
cli.utils.train.launch_training(
cfg_file,
launcher,
cloud,
kwargs,=None,
launcher_args=False,
use_exec )
Execute training with the given configuration.