Skip to content
Nodus

nodus.integrations.lightning

View Markdown

NodusCallback for a Lightning Trainer (ADR-103).

from nodus.integrations.lightning import NodusCallback


trainer = L.Trainer(callbacks=[NodusCallback()], default_root_dir=nodus.state_dir())
trainer.fit(model, ckpt_path=NodusCallback.last_checkpoint())

NodusCallback

Section titled “NodusCallback”
class NodusCallback(every_n_steps: int = 10, phase: str = 'Training') -> None

Metrics from trainer.callback_metrics, progress, phases and the checkpoint handshake.

NodusCallback.last_checkpoint

Section titled “NodusCallback.last_checkpoint”
last_checkpoint(directory: str | os.PathLike[str] | None = None) -> str | None

The checkpoint this callback saved in an earlier attempt, for trainer.fit(ckpt_path=...).

NodusCallback.on_train_batch_end

Section titled “NodusCallback.on_train_batch_end”
on_train_batch_end(trainer: Any, pl_module: Any, outputs: Any, batch: Any, batch_idx: int) -> None

NodusCallback.on_train_start

Section titled “NodusCallback.on_train_start”
on_train_start(trainer: Any, pl_module: Any) -> None