struct

ML::Optim::ParamGroup

Inherits Struct / Value / Object

Parameter group for different learning rates

Constructors

new(params : Array(Autograd::Variable), lr : Float32 = 0.001_f32, weight_decay : Float32 = 0.0_f32, name : String = "default")
Source

Instance methods

lr=(lr : Float32)
Source
name
Source
name=(name : String)
Source
params
Source
params=(params : Array(Autograd::Variable))
Source
weight_decay
Source
weight_decay=(weight_decay : Float32)
Source