Trainer
=======

.. currentmodule:: lib.training.train

.. autoclass:: Trainer
   :show-inheritance:

   .. rubric:: Attributes Summary

   .. autosummary::

      ~Trainer.exit_early

   .. rubric:: Methods Summary

   .. autosummary::

      ~Trainer.save
      ~Trainer.toggle_mask
      ~Trainer.train_one_batch
      ~Trainer.train_one_step

   .. rubric:: Attributes Documentation

   .. autoattribute:: exit_early

   .. rubric:: Methods Documentation

   .. automethod:: save
   .. automethod:: toggle_mask
   .. automethod:: train_one_batch
   .. automethod:: train_one_step