pytorch\_tao.plugins.TrainPlugin
================================

.. currentmodule:: pytorch_tao.plugins

.. autoclass:: TrainPlugin

   
   .. automethod:: __init__

   
   .. rubric:: Methods

   .. autosummary::
   
      ~TrainPlugin.__init__
      ~TrainPlugin.after_use
      ~TrainPlugin.attach
      ~TrainPlugin.set_engine
   
   

   
   
   .. rubric:: Attributes

   .. autosummary::
   
      ~TrainPlugin.engine
      ~TrainPlugin.trainer