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(merge_hash = nil) ⇒ `Object`

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

def to_hash(merge_hash = nil)
  hash = {class: self.class.name}
  hash.merge!(merge_hash) if merge_hash
  hash
end