ksuit.utils.model.helpers
=========================

.. py:module:: ksuit.utils.model.helpers


Functions
---------

.. autoapisummary::

   ksuit.utils.model.helpers.compute_model_norm


Module Contents
---------------

.. py:function:: compute_model_norm(module)

   Compute the norm of the weights of a nn.Module, useful to check if weight initialization has been correctly
   applied (i.e., the model weights has changed)

   :param module: nn.Module to compute the norm for.