ksuit.callbacks.checkpoint_callbacks.checkpoint
===============================================

.. py:module:: ksuit.callbacks.checkpoint_callbacks.checkpoint


Classes
-------

.. autoapisummary::

   ksuit.callbacks.checkpoint_callbacks.checkpoint.CheckpointCallback


Module Contents
---------------

.. py:class:: CheckpointCallback(callback_config, **kwargs)

   Bases: :py:obj:`ksuit.callbacks.base.periodic_callback.PeriodicCallback`


   Callback to save the model and optimizer state periodically.

   Initializes the CheckpointCallback.

   :param callback_config: The configuration for the callback. Implements the CheckpointCallbackConfig schema.
   :param \*\*kwargs: additional arguments passed to the parent class.


   .. py:attribute:: save_weights


   .. py:attribute:: save_optim


   .. py:attribute:: save_latest_weights


   .. py:attribute:: save_latest_optim


   .. py:attribute:: model_names
      :value: []