ksuit.callbacks.offline_callbacks.offline_loss

Classes

OfflineLossCallback

A periodic Callback that is invoked at the end of each epoch to calculate and track the loss and a dataset.

Module Contents

class ksuit.callbacks.offline_callbacks.offline_loss.OfflineLossCallback(callback_config, **kwargs)

Bases: ksuit.callbacks.base.PeriodicIteratorCallback

A periodic Callback that is invoked at the end of each epoch to calculate and track the loss and a dataset.

Initializes the OfflineLossCallback.

Parameters:

callback_config (ksuit.schemas.callbacks.callbacks_config.OfflineLossCallbackConfig) – configuration of the OfflineLossCallback. Implements the OfflineLossCallbackConfig schema.

dataset_key
output_patterns_to_log