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.


Learn more about Qdrant vector search project and ecosystem

Discover Qdrant

Similarity Learning

Explore practical problem solving with Similarity Learning

Learn Similarity Learning


Find people dealing with similar problems and get answers to your questions

Join Community