network
=======

.. List the submodules





.. Autodoc anything defined in the module itself

   TODO: WE DON'T USE THIS OPTION RIGHT NOW BUT WE CAN REACTIVATE IF WANTED
   We use :ignore-module-all: so sphinx does not document the same module twice, even if it is reimported
   For reimports that should be documented somewhere other than where they are defined, the re-imports
   __module__ should be manually overridden -- i.e. in the ``__init__.py`` which contains ``from xxx import YYY``,
   add in ``YYY.__module__ = __name__``.

.. automodule:: modelopt.torch.utils.network
   :members:
   :undoc-members:

   .. Also show members without docstrings. Only members from __all__ are considered as per conf.py
   .. Ideally we should add docstrings for these members.


   .. Overview table of available classes in the module
   
   
   


   .. Overview table of available functions in the module
   
   
   .. rubric:: Functions

   .. autosummary::
      :nosignatures:
   
      compare_dict
      get_model_attributes
      get_module_device
      get_same_padding
      init_model_from_model_like
      is_channels_last
      is_parallel
      make_divisible
      model_to
      param_num
      param_num_from_forward
      remove_bn
      run_forward_loop
      set_submodule
      standardize_model_args
      standardize_model_like_tuple
      standardize_named_model_args
      standardize_constructor_args
      unwrap_model
      zero_grad
      create_param_grad_clear_hook
      get_unwrapped_name