noether.core.callbacks.default.param_count

Classes

ParamCountCallback

Callback to log the number of trainable and frozen parameters of the model.

Module Contents

class noether.core.callbacks.default.param_count.ParamCountCallback(trainer, model, data_container, tracker, log_writer, checkpoint_writer, metric_property_provider, name=None)

Bases: noether.core.callbacks.base.CallbackBase

Callback to log the number of trainable and frozen parameters of the model.

Parameters:

  • trainer (noether.training.trainers.BaseTrainer) – Trainer of the current run, subclass of SgdTrainer.

  • model (noether.core.models.ModelBase) – Model of the current run.

  • data_container (noether.data.container.DataContainer) – DataContainer instance that provides access to all datasets.

  • tracker (noether.core.trackers.BaseTracker) – Tracker instance to log metrics to stdout/disk/online platform.

  • log_writer (noether.core.writers.LogWriter) – LogWriter instance to log metrics to stdout/disk/online platform.

  • metric_property_provider (noether.core.providers.metric_property.MetricPropertyProvider) – MetricPropertyProvider instance to access properties of metrics.

  • name (str | None) – Name of the callback.

  • checkpoint_writer (noether.core.writers.CheckpointWriter)