ksuit.callbacks.default_callbacks.train_time

Classes

TrainTimeCallback

Callback to log the time spent on dataloading during training.

Module Contents

class ksuit.callbacks.default_callbacks.train_time.TrainTimeCallback(callback_config, **kwargs)

Bases: ksuit.callbacks.base.PeriodicCallback

Callback to log the time spent on dataloading during training.

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.

train_data_times = []
total_train_data_time = 0.0