eval.py

Model Overview

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

Architecture

  • Architecture: tiny transformer
  • Scale: base
  • Attention: flash
  • Fusion strategy: tensor fusion
  • Task head: contrastive
  • Activation: gelu tanh
  • Normalization: layernorm
  • Initialization: kaiming normal

Training

  • Optimizer: novograd
  • LR scheduler: cosine

Files

  • eval.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