Class: DNN::Models::FixedModel

Inherits:
Model show all
Defined in:
lib/dnn/core/models.rb

Instance Attribute Summary collapse

Attributes inherited from Model

#last_log, #loss_weights, #optimizer

Instance Method Summary collapse

Methods inherited from Model

#add_callback, #built?, #call, #clean_layers, #clear_callbacks, #copy, #evaluate, #evaluate_by_iterator, #get_all_params_data, #get_layer, load, #load_params, #loss_func, #loss_func=, #predict, #predict1, #save, #save_params, #set_all_params_data, #setup, #test_on_batch, #train, #train_by_iterator, #train_on_batch, #trainable_layers

Methods inherited from Chain

#call, #load_hash, #to_hash

Constructor Details

#initialize(output_tensor, layers) ⇒ FixedModel

Returns a new instance of FixedModel.



658
659
660
661
662
 
# File 'lib/dnn/core/models.rb', line 658

def initialize(output_tensor, layers)
  super()
  @input_link = get_input_link(output_tensor.link)
  @layers = layers
end

Instance Attribute Details

#layers ⇒ Object (readonly)

Returns the value of attribute layers.



656
657
658
 
# File 'lib/dnn/core/models.rb', line 656

def layers
  @layers
end

Instance Method Details

#forward(input_tensors) ⇒ Object 



664
665
666
667
668
669
670
671
672
 
# File 'lib/dnn/core/models.rb', line 664

def forward(input_tensors)
  if input_tensors.is_a?(Array)
    input_tensors.each do |tensor|
      @input_link.forward(tensor)
    end
  else
    @input_link.forward(input_tensors)
  end
end