dymad.training.trainer_run¶

Classes

TrainerRun(config, model_class, device, ...)

Owns one concrete training run identity, artifacts, and typed phase pipeline.

class dymad.training.trainer_run.TrainerRun(config, model_class, device, dtype, run_name, checkpoint_prefix, results_prefix, execution_services=None)¶

Bases: object

Owns one concrete training run identity, artifacts, and typed phase pipeline.

load_run_checkpoint(path=None)¶

Return type:: tuple[TrainerState, ArtifactRegistry]

run(*, initial_context, initial_state=None, artifacts=None)¶

Return type:: list[PhaseResult]

property run_checkpoint_path: str¶

save_run_checkpoint(trainer_state, artifacts)¶

Return type:: str