Module: Transformers::Vit

Defined in:

lib/transformers/models/vit/modeling_vit.rb,
lib/transformers/models/vit/configuration_vit.rb,
lib/transformers/models/vit/image_processing_vit.rb

Defined Under Namespace

Classes: ViTAttention, ViTConfig, ViTEmbeddings, ViTEncoder, ViTForImageClassification, ViTImageProcessor, ViTIntermediate, ViTLayer, ViTModel, ViTOutput, ViTPatchEmbeddings, ViTPooler, ViTPreTrainedModel, ViTSelfAttention, ViTSelfOutput

Constant Summary

VIT_ATTENTION_CLASSES =

{
  "eager" => ViTAttention
}