noether.core.callbacks.default.train_time

Classes

TrainTimeCallback

Callback to log the time spent on dataloading during training.

Module Contents

class noether.core.callbacks.default.train_time.TrainTimeCallback(callback_config, **kwargs)

Bases: noether.core.callbacks.periodic.PeriodicCallback

Callback to log the time spent on dataloading during training.

Initializes the PeriodicCallback.

Parameters:

  • callback_config (noether.core.schemas.callbacks.CallBackBaseConfig) – 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: list[float] = []
total_train_data_time = 0.0