Class: Informers::ViTForImageClassification
- Inherits:
-
ViTPreTrainedModel
- Object
- PreTrainedModel
- ViTPreTrainedModel
- Informers::ViTForImageClassification
- Defined in:
- lib/informers/models.rb
Constant Summary
Constants inherited from PreTrainedModel
PreTrainedModel::MAIN_INPUT_NAME
Instance Attribute Summary
Attributes inherited from PreTrainedModel
Instance Method Summary collapse
Methods inherited from PreTrainedModel
construct_session, from_pretrained, #generate, #initialize
Constructor Details
This class inherits a constructor from Informers::PreTrainedModel
Instance Method Details
#call(model_inputs) ⇒ Object
961 962 963 |
# File 'lib/informers/models.rb', line 961 def call(model_inputs) SequenceClassifierOutput.new(*super(model_inputs)) end |