ksuit.callbacks.default_callbacks.param_count
=============================================

.. py:module:: ksuit.callbacks.default_callbacks.param_count


Classes
-------

.. autoapisummary::

   ksuit.callbacks.default_callbacks.param_count.ParamCountCallback


Module Contents
---------------

.. py:class:: ParamCountCallback(trainer, model, data_container, tracker, log_writer, checkpoint_writer, metric_property_provider, name = None)

   Bases: :py:obj:`ksuit.callbacks.base.callback_base.CallbackBase`


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

   :param trainer: Trainer of the current run, subclass of `SgdTrainer`.
   :param model: Model of the current run.
   :param data_container: DataContainer instance that provides access to all datasets.
   :param tracker: Tracker instance to log metrics to stdout/disk/online platform.
   :param log_writer: LogWriter instance to log metrics to stdout/disk/online platform.
   :param metric_property_provider: MetricPropertyProvider instance to access properties of metrics.
   :param name: Name of the callback.