quaterion.train.callbacks.cleanup_callback module¶

class CleanupCallback[source]¶

Bases: Callback

teardown(trainer: Trainer, pl_module: TrainableModel, stage: str | None = None) → None[source]¶

Called when fit, validate, test, predict, or tune ends.