Rodriguezsen
/

mobilenet-recsys-demo

Model card Files Files and versions

inference.py

Model Overview

A tiny-scale implementation of the albef architecture, built for retrieval tasks.

Architecture

Architecture: albef
Scale: tiny
Attention: grouped query
Fusion strategy: co attention
Task head: retrieval
Activation: swish
Normalization: batchnorm
Initialization: kaiming normal

Training

Optimizer: rmsprop
LR scheduler: exponential

Files

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