schnetpack.train
================
.. currentmodule:: train


Callbacks
---------

.. autosummary::
    :toctree: generated
    :nosignatures:
    :template: classtemplate.rst

    ModelCheckpoint
    PredictionWriter

Scheduler
---------

.. autosummary::
    :toctree: generated
    :nosignatures:
    :template: classtemplate.rst

    ReduceLROnPlateau

