emmi.functional.init
====================

.. py:module:: emmi.functional.init


Attributes
----------

.. autoapisummary::

   emmi.functional.init.ALL_CONVS
   emmi.functional.init.ALL_LAYERS


Functions
---------

.. autoapisummary::

   emmi.functional.init.init_bias_to_zero
   emmi.functional.init.init_trunc_normal_zero_bias


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

.. py:data:: ALL_CONVS

.. py:data:: ALL_LAYERS

.. py:function:: init_bias_to_zero(layer_module)

   Initialize the bias tensor of a nn.Module instance to zero.

   :param layer_module: An nn.Module instance, either a Linear or Conv layer.


.. py:function:: init_trunc_normal_zero_bias(layer_module, std = 0.02)

   Initialize the weight tensor of a nn.Module instance using the truncated normal initialization with a zero bias vector.

   :param layer_module: An nn.Module instance, either a Linear or Conv layer.
   :param std: Standard Deviation value of the normal distribution to sample weights from. Defaults to 0.02.