Class: DNN::Initializers::He

Inherits:

Initializer

• Object
• Initializer
• DNN::Initializers::He

Defined in:

lib/dnn/core/initializers.rb

Instance Method Summary

• #init_param(layer, param_key) ⇒ Object

Methods inherited from Initializer

#to_hash

Instance Method Details

#init_param(layer, param_key) ⇒ Object

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

def init_param(layer, param_key)
  num_prev_nodes = layer.prev_layer.shape.reduce(:*)
  super(layer, param_key, layer.params[param_key].rand_norm / Math.sqrt(num_prev_nodes) * Math.sqrt(2))
end