WrappedModel
============

.. currentmodule:: plugins.train.trainer.distributed

.. autoclass:: WrappedModel
   :show-inheritance:

   .. rubric:: Attributes Summary

   .. autosummary::

      ~WrappedModel.T_destination
      ~WrappedModel.call_super_init
      ~WrappedModel.dump_patches

   .. rubric:: Methods Summary

   .. autosummary::

      ~WrappedModel.__call__
      ~WrappedModel.add_module
      ~WrappedModel.apply
      ~WrappedModel.bfloat16
      ~WrappedModel.buffers
      ~WrappedModel.children
      ~WrappedModel.compile
      ~WrappedModel.cpu
      ~WrappedModel.cuda
      ~WrappedModel.double
      ~WrappedModel.eval
      ~WrappedModel.extra_repr
      ~WrappedModel.float
      ~WrappedModel.forward
      ~WrappedModel.get_buffer
      ~WrappedModel.get_extra_state
      ~WrappedModel.get_parameter
      ~WrappedModel.get_submodule
      ~WrappedModel.half
      ~WrappedModel.ipu
      ~WrappedModel.load_state_dict
      ~WrappedModel.modules
      ~WrappedModel.mtia
      ~WrappedModel.named_buffers
      ~WrappedModel.named_children
      ~WrappedModel.named_modules
      ~WrappedModel.named_parameters
      ~WrappedModel.parameters
      ~WrappedModel.register_backward_hook
      ~WrappedModel.register_buffer
      ~WrappedModel.register_forward_hook
      ~WrappedModel.register_forward_pre_hook
      ~WrappedModel.register_full_backward_hook
      ~WrappedModel.register_full_backward_pre_hook
      ~WrappedModel.register_load_state_dict_post_hook
      ~WrappedModel.register_load_state_dict_pre_hook
      ~WrappedModel.register_module
      ~WrappedModel.register_parameter
      ~WrappedModel.register_state_dict_post_hook
      ~WrappedModel.register_state_dict_pre_hook
      ~WrappedModel.requires_grad_
      ~WrappedModel.set_extra_state
      ~WrappedModel.set_submodule
      ~WrappedModel.share_memory
      ~WrappedModel.state_dict
      ~WrappedModel.to
      ~WrappedModel.to_empty
      ~WrappedModel.train
      ~WrappedModel.type
      ~WrappedModel.xpu
      ~WrappedModel.zero_grad

   .. rubric:: Attributes Documentation

   .. autoattribute:: T_destination
   .. autoattribute:: call_super_init
   .. autoattribute:: dump_patches

   .. rubric:: Methods Documentation

   .. automethod:: __call__
   .. automethod:: add_module
   .. automethod:: apply
   .. automethod:: bfloat16
   .. automethod:: buffers
   .. automethod:: children
   .. automethod:: compile
   .. automethod:: cpu
   .. automethod:: cuda
   .. automethod:: double
   .. automethod:: eval
   .. automethod:: extra_repr
   .. automethod:: float
   .. automethod:: forward
   .. automethod:: get_buffer
   .. automethod:: get_extra_state
   .. automethod:: get_parameter
   .. automethod:: get_submodule
   .. automethod:: half
   .. automethod:: ipu
   .. automethod:: load_state_dict
   .. automethod:: modules
   .. automethod:: mtia
   .. automethod:: named_buffers
   .. automethod:: named_children
   .. automethod:: named_modules
   .. automethod:: named_parameters
   .. automethod:: parameters
   .. automethod:: register_backward_hook
   .. automethod:: register_buffer
   .. automethod:: register_forward_hook
   .. automethod:: register_forward_pre_hook
   .. automethod:: register_full_backward_hook
   .. automethod:: register_full_backward_pre_hook
   .. automethod:: register_load_state_dict_post_hook
   .. automethod:: register_load_state_dict_pre_hook
   .. automethod:: register_module
   .. automethod:: register_parameter
   .. automethod:: register_state_dict_post_hook
   .. automethod:: register_state_dict_pre_hook
   .. automethod:: requires_grad_
   .. automethod:: set_extra_state
   .. automethod:: set_submodule
   .. automethod:: share_memory
   .. automethod:: state_dict
   .. automethod:: to
   .. automethod:: to_empty
   .. automethod:: train
   .. automethod:: type
   .. automethod:: xpu
   .. automethod:: zero_grad