ksuit.callbacks.online_callbacks.update_output

Classes

UpdateOutputCallback

Callback that is invoked during training after every gradient step to track certain outputs from the update step.

Module Contents

class ksuit.callbacks.online_callbacks.update_output.UpdateOutputCallback(callback_config, **kwargs)

Bases: ksuit.callbacks.base.PeriodicCallback

Callback that is invoked during training after every gradient step to track certain outputs from the update step.

The provided update_outputs are assumed to be a dictionary and outputs that match keys or patterns are tracked. An update output matches if either the key matches exactly, e.g. {“some_output”: …} and keys[“some_output”]; or if one of the patterns is contained in the update key name, e.g. {“some_loss”: …} and patterns = [“loss”].

Initializes the UpdateOutputCallback class.

Parameters:

• callback_config (ksuit.schemas.callbacks.callbacks_config.UpdateOutputCallbackConfig) – The configuration for the callback. Implements the UpdateOutputCallbackConfig schema.

• **kwargs – additional keyword arguments provided to the parent cla

patterns

keys

verbose

tracked_values

reduce

log_output

save_output