trainerERA5_v2
==============

.. py:module:: trainerERA5_v2

.. autoapi-nested-parse::

   trainer.py
   -------------------------------------------------------
   Content:
       - Trainer
           - train_one_epoch
           - validate
           - fit



Attributes
----------

.. autoapisummary::

   trainerERA5_v2.logger


Classes
-------

.. autoapisummary::

   trainerERA5_v2.Trainer


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

.. py:data:: logger

.. py:class:: Trainer(model: torch.nn.Module, rank: int)

   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, conf, trainloader, optimizer, criterion, scaler, scheduler, metrics)


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


   .. py:method:: fit_deprecated(conf, train_loader, valid_loader, optimizer, train_criterion, valid_criterion, scaler, scheduler, metrics, rollout_scheduler=None, trial=False)