pub enum Init {
Const(f64),
Randn {
mean: f64,
stdev: f64,
},
Uniform {
lo: f64,
up: f64,
},
Kaiming {
dist: NormalOrUniform,
fan: FanInOut,
non_linearity: NonLinearity,
},
Orthogonal {
gain: f64,
},
}
Expand description
Variable initializations.
Variants§
Const(f64)
Constant value.
Randn
Random normal with some mean and standard deviation.
Uniform
Uniform initialization between some lower and upper bounds.
Kaiming
Kaiming uniform initialization. See “Delving deep into rectifiers: Surpassing human-level performance on ImageNet classification” He, K. et al. (2015). This uses a uniform distribution.
Orthogonal
Orthogonal initialization