.. Custom template for autosummary modules. Can be removed when
.. https://github.com/sphinx-doc/sphinx/issues/7912 is fixed.

dgs.utils.torchtools
====================

.. automodule:: dgs.utils.torchtools

   
       
   

   
       
           .. rubric:: Module Functions

           .. autosummary::
              :toctree:
              :template: function.rst
           
              close_all_layers
              close_specified_layers
              configure_torch_module
              get_model_from_module
              init_instance_params
              init_model_params
              load_checkpoint
              load_pretrained_weights
              open_all_layers
              open_specified_layers
              resume_from_checkpoint
              save_checkpoint
              set_bn_to_eval
              torch_memory_analysis
              torch_memory_analysis_win