credit.trainers.trainer_downscaling
===================================

.. py:module:: credit.trainers.trainer_downscaling


Attributes
----------

.. autoapisummary::

   credit.trainers.trainer_downscaling.logger


Classes
-------

.. autoapisummary::

   credit.trainers.trainer_downscaling.TrainerDownscaling


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

.. py:data:: logger

.. py:class:: TrainerDownscaling(model: torch.nn.Module, rank: int, conf: dict)

   Bases: :py:obj:`credit.trainers.base_trainer.BaseTrainer`


   Helper class that provides a standard way to create an ABC using
   inheritance.


   .. py:method:: setup(conf)


   .. py:method:: train_one_epoch(epoch, conf, trainloader, optimizer, criterion, scaler, scheduler, metrics)


   .. py:method:: validate(epoch, valid_loader, criterion, metrics)

      Validates the model on the validation dataset.

      :param epoch: Current epoch number.
      :type epoch: int
      :param conf: Configuration dictionary containing validation settings.
      :type conf: dict
      :param valid_loader: DataLoader for the validation dataset.
      :type valid_loader: DataLoader
      :param criterion: Loss function used for validation.
      :type criterion: callable
      :param metrics: Function to compute metrics for evaluation.
      :type metrics: callable

      :returns: Dictionary containing validation metrics and loss for the epoch.
      :rtype: dict