walkertyler96
/

resnet-classify40

cross-attention

Model card Files Files and versions

finetune.py

Model Overview

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

Architecture

Architecture: mobilevit
Scale: huge
Attention: sparse
Fusion strategy: cross attention
Task head: retrieval
Activation: gelu
Normalization: groupnorm
Initialization: kaiming

Training

Optimizer: sgd
LR scheduler: polynomial

Files

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