noether.training.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 noether.training.callbacks.offline_loss.OfflineLossCallback(callback_config, **kwargs)

Bases: noether.training.callbacks.periodic_iterator.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 (noether.core.schemas.callbacks.OfflineLossCallbackConfig) – configuration of the OfflineLossCallback. Implements the OfflineLossCallbackConfig schema.

dataset_key

output_patterns_to_log