ksuit.callbacks.default_callbacks.online_loss¶

Classes¶

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

class ksuit.callbacks.default_callbacks.online_loss.OnlineLossCallback(callback_config, **kwargs)¶

Bases: 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.

Parameters:

callback_config (ksuit.schemas.callbacks.callbacks_config.OnlineLossCallbackConfig | None) – The configuration for the callback.
**kwargs – additional arguments passed to the parent class.