Callbacks
=======================

.. automodule:: torchtnt.framework.callback
   :members:
   :undoc-members:


Built-in callbacks
~~~~~~~~~~~~~~~~~~~~~

We offer several pre-written callbacks which are ready to be used out of the box:


.. currentmodule:: torchtnt.framework.callbacks

.. autosummary::
    :nosignatures:
    :toctree: generated/
    :template: class_template.rst

    BaseCSVWriter
    EarlyStopping
    GarbageCollector
   IterationTimeLogger
    Lambda
    LearningRateMonitor
    MemorySnapshot
    ModuleSummary
    PeriodicDistributedSync
    ProgressReporter
    PyTorchProfiler
    SlowRankDetector
    SystemResourcesMonitor
    TensorBoardParameterMonitor
    TimeLimitInterrupter
    TimeWaitForBatchLogger
    ThroughputLogger
    TorchSnapshotSaver
    TQDMProgressBar
    TrainProgressMonitor
