modelzoo.common.pytorch.model_utils.weight_initializers#

Functions

lecun_normal_

Adapted from TensorFlow's initializations https://www.tensorflow.org/api_docs/python/tf/keras/initializers/LecunNormal

lecun_uniform_

Adapted from TensorFlow's initializations https://www.tensorflow.org/api_docs/python/tf/keras/initializers/LecunUniform

trunc_normal_

Fills the input Tensor with values drawn from a truncated normal distribution.

variance_scaling_

Adapted from TensorFlow's initializations https://www.tensorflow.org/api_docs/python/tf/keras/initializers/VarianceScaling