class Num::NN::SGDMomentumOptimizer(T)

Defined in:

nn/optimizers/sgd.cr

Constructors

Instance Method Summary

Instance methods inherited from class Num::NN::Optimizer(T)

build_params(l : Array(Layer(T))) build_params, learning_rate : Float64 learning_rate, params : Array(Num::Grad::Variable(T)) params, update update

Constructor methods inherited from class Num::NN::Optimizer(T)

new(learning_rate : Float64 = 0.01) new

Constructor Detail

def self.new(learning_rate : Float64, momentum : Float64 = 0.0, decay : Float64 = 0.0, nesterov : Bool = false) #

[View source]

Instance Method Detail

def build_params(l : Array(Layer(T))) #

[View source]
def decay : Float64 #

[View source]
def learning_rate : Float64 #

[View source]
def moments : Array(T) #

[View source]
def momentum : Float64 #

[View source]
def nesterov : Bool #

[View source]
def params : Array(Num::Grad::Variable(T)) #

[View source]
def update #

[View source]