espnet2.torch_utils.initialize.initialize

source

espnet2.torch_utils.initialize.initialize(model: Module, init: str)

Initialize weights of a neural network module.

Parameters are initialized using the given method or distribution.

Custom initialization routines can be implemented into submodules as function espnet_initialization_fn within the custom module.

• Parameters:
  • model – Target.
  • init – Method of initialization.