Method: Informers::PreTrainedModel#initialize

Defined in:
lib/informers/models.rb

#initialize(config, session) ⇒ PreTrainedModel

Returns a new instance of PreTrainedModel.



66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
 
# File 'lib/informers/models.rb', line 66

def initialize(config, session)
  super()

  @config = config
  @session = session

  @output_names = nil

  model_name = MODEL_CLASS_TO_NAME_MAPPING[self.class]
  model_type = MODEL_TYPE_MAPPING[model_name]

  case model_type
  when MODEL_TYPES[:DecoderOnly]
    raise Todo
  when MODEL_TYPES[:Seq2Seq], MODEL_TYPES[:Vision2Seq]
    raise Todo
  when MODEL_TYPES[:EncoderDecoder]
    raise Todo
  else
    @forward = method(:encoder_forward)
  end
end