Package ai.djl.training.initializer
package ai.djl.training.initializer
Contains classes for initializing a
Block
's Parameter
s.
It contains a main interface Initializer
and various
initializers that extend it.
-
ClassDescriptionInitializer that generates tensors with constant values.An interface representing an initialization method.
NormalInitializer
initializes weights with random values sampled from a normal distribution with a mean of zero and standard deviation ofsigma
.Naive implementation of a truncated normal initializer.UniformInitializer
initializes weights with random values uniformly sampled from a given range.XavierInitializer
is anInitializer
that performs "Xavier" initialization for parameters.Enum for different types of factor type.Enum for different types of random distributions.