dymad.training.execution_services¶

Classes

ExecutionServices(device, checkpoint_prefix, ...)

Non-checkpointable runtime policy for one training execution path.

class dymad.training.execution_services.ExecutionServices(device, checkpoint_prefix, results_prefix, log_level='info', log_stdout=False)¶

Bases: object

Non-checkpointable runtime policy for one training execution path.

apply_to_config(config)¶

Return type:: dict[str, Any]

checkpoint_file(file_name)¶

Return type:: str

checkpoint_prefix: str¶

configure_logger(name, prefix=None)¶

Return type:: Logger

device: device¶

ensure_artifact_dirs()¶

Return type:: None

classmethod from_config(config, default_device=None)¶

Return type:: ExecutionServices

classmethod from_driver_config(base_config, config_path, default_device=None)¶

Return type:: ExecutionServices

log_level: str = 'info'¶

log_stdout: bool = False¶

logger_prefix(default_name)¶

Return type:: str

results_prefix: str¶

with_device(device)¶

Return type:: ExecutionServices

with_paths(*, checkpoint_prefix=None, results_prefix=None)¶

Return type:: ExecutionServices