marchenkoiryna
/

hw1-classify

Model card Files Files and versions

predict.py

Model Overview

A xlarge-scale implementation of the mobilevit architecture, built for retrieval tasks.

Architecture

Architecture: mobilevit
Scale: xlarge
Attention: sliding window
Fusion strategy: gated fusion
Task head: retrieval
Activation: relu
Normalization: batchnorm
Initialization: xavier

Training

Optimizer: lion
LR scheduler: cosine

Files

predict.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