ksuit.callbacks.default_callbacks.online_loss
=============================================

.. py:module:: ksuit.callbacks.default_callbacks.online_loss


Classes
-------

.. autoapisummary::

   ksuit.callbacks.default_callbacks.online_loss.OnlineLossCallback


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

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

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


   Callback to track the loss of the model after every gradient accumulation step and log the average loss.

   Initializes the OnlineLossCallback.

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


   .. py:attribute:: verbose


   .. py:attribute:: tracked_losses
      :type:  collections.defaultdict[str, list[torch.Tensor]]