train.py
Model Overview
A small-scale implementation of the mobilevit architecture, built for contrastive tasks.
Architecture
- Architecture: mobilevit
- Scale: small
- Attention: sliding window
- Fusion strategy: bilinear
- Task head: contrastive
- Activation: mish
- Normalization: batchnorm
- Initialization: xavier uniform
Training
- Optimizer: adafactor
- LR scheduler: onecycle
Files
train.py— main artifact of this repository
License
See the license field above.
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support