dymad.training.phase_pipeline

Classes

PhasePipeline(config, model_class, device, dtype)

Runs typed training phases in sequence.

class dymad.training.phase_pipeline.PhasePipeline(config, model_class, device, dtype, execution_services=None)

Bases: object

Runs typed training phases in sequence.

build_phase(spec)

run(*, initial_context, initial_state, artifacts=None, run_name, checkpoint_callback=None)

Return type:

list[PhaseResult]