Class: DNN::Initializers::Initializer

Inherits:

Object

Object
DNN::Initializers::Initializer

show all

Defined in:: lib/dnn/core/initializers.rb

Direct Known Subclasses

He, RandomNormal, Xavier, Zeros

Instance Method Summary collapse

Instance Method Details

#init_param(layer, param_key, param) ⇒ `Object`



5
6
7

# File 'lib/dnn/core/initializers.rb', line 5

def init_param(layer, param_key, param)
  layer.params[param_key] = param
end

#to_hash(hash) ⇒ `Object`



9
10
11

# File 'lib/dnn/core/initializers.rb', line 9

def to_hash(hash)
  {name: self.class.name}.merge(hash)
end