quaterion.train.callbacks.cleanup_callback 模块

class CleanupCallback

基类: Callback

teardown(trainer: Trainer, pl_module: TrainableModel, stage: str | None = None) → None

在 fit, validate, test, predict, 或 tune 结束时调用。