arjunpweu
/

unet-summarizer

Model card Files Files and versions

pipeline.py

Model Overview

A base-scale implementation of the deit architecture, built for contrastive tasks.

Architecture

Architecture: deit
Scale: base
Attention: linear
Fusion strategy: tucker
Task head: contrastive
Activation: approx gelu
Normalization: groupnorm
Initialization: kaiming normal

Training

Optimizer: lion
LR scheduler: onecycle

Files

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