mighty.trainer.MaskTrainer
==========================

.. currentmodule:: mighty.trainer

.. autoclass:: MaskTrainer
   :members:
   :inherited-members:
   :special-members: __call__

   

   
   .. rubric:: Methods

   .. autosummary::
   
      ~MaskTrainer.__init__
      ~MaskTrainer.cpu
      ~MaskTrainer.cuda
      ~MaskTrainer.get_probability
      ~MaskTrainer.train_mask
   
   

   
   
   .. rubric:: Attributes

   .. autosummary::
   
      ~MaskTrainer.l1_coeff
      ~MaskTrainer.mask_size
      ~MaskTrainer.max_iterations
      ~MaskTrainer.tv_beta
      ~MaskTrainer.tv_coeff