core.MEDIAR.Trainer
===================

.. py:module:: core.MEDIAR.Trainer




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

.. py:class:: Trainer(model, dataloaders, optimizer, scheduler=None, criterion=None, num_epochs=100, device='cuda:0', no_valid=False, valid_frequency=1, amp=False, algo_params=None)

   Bases: :py:obj:`core.BaseTrainer.BaseTrainer`


   Abstract base class for trainer implementations


   .. py:attribute:: mse_loss


   .. py:attribute:: bce_loss


   .. py:method:: mediar_criterion(outputs, labels_onehot_flows)

      loss function between true labels and prediction outputs