credit.trainers.trainerWRF
==========================

.. py:module:: credit.trainers.trainerWRF


Attributes
----------

.. autoapisummary::

   credit.trainers.trainerWRF.logger


Classes
-------

.. autoapisummary::

   credit.trainers.trainerWRF.TrainerWRF


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

.. py:data:: logger

.. py:class:: TrainerWRF(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:: train_one_epoch(epoch, trainloader, optimizer, criterion, scaler, scheduler, metrics)


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