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

previous

modelzoo.common.pytorch.model_utils.vocab_utils.get_vocab_size

next

modelzoo.common.pytorch.model_utils.weight_initializers.lecun_normal_