Ananyamenon
/

tokenizer

cross-attention

Model card Files Files and versions

model.py

Model Overview

A nano-scale implementation of the vit architecture, built for contrastive tasks.

Architecture

Architecture: vit
Scale: nano
Attention: linear
Fusion strategy: cross attention
Task head: contrastive
Activation: gelu
Normalization: batchnorm
Initialization: kaiming normal

Training

Optimizer: adamw
LR scheduler: step

Files

model.py — main artifact of this repository

License

See the license field above.

Downloads last month: -; Downloads are not tracked for this model. How to track

Inference Providers NEW

This model isn't deployed by any Inference Provider. 🙋 Ask for provider support