nntoolbox.init.normal module

nntoolbox.init.normal.normal_init(module: torch.nn.modules.module.Module, mean: float, std: float)

Initialize the weight of a module to normal distribution of given mean and std If module has bias, assign it to zero constant (UNTESTED)

Parameters

module – must have weight tensor

Param

mean of distribution

Param

std: standard deviation of the distribution