ksuit.callbacks.default_callbacks.progress

Classes

ProgressCallback

Callback to print the progress of the training such as number of epochs and updates.

Module Contents

class ksuit.callbacks.default_callbacks.progress.ProgressCallback(callback_config, **kwargs)

Bases: ksuit.callbacks.base.PeriodicCallback

Callback to print the progress of the training such as number of epochs and updates.

Initializes the PeriodicCallback.

Parameters:

  • callback_config (ksuit.schemas.callbacks.callbacks_config.CallBackBaseConfig | None) – Configuration of the PeriodicCallback. Implements the CallBackBaseConfig schema.

  • trainer – Trainer of the current run, subclass of SgdTrainer.

  • model – Model of the current run.

  • data_container – DataContainer instance that provides access to all datasets.

  • tracker – Tracker instance to log metrics to stdout/disk/online platform.

  • log_writer – LogWriter instance to log metrics.

  • checkpoint_writer – CheckpointWriter instance to save checkpoints.

  • metric_property_provider – MetricPropertyProvider instance to access properties of metrics.

  • name – Name of the callback.