Akashraobury's picture
Upload 2 files
22e4917 verified
|
Raw
History Blame Contribute Delete
772 Bytes
metadata
license: apache-2.0
tags:
  - adam
  - flamingo
  - gelu
  - linear-warmup
  - low-rank
  - nano
  - orthogonal
  - retrieval
  - scalenorm
  - standard

model_112034448_flamingo_nano.py

Model Overview

A nano-scale implementation of the flamingo architecture, built for retrieval tasks.

Architecture

  • Architecture: flamingo
  • Scale: nano
  • Attention: standard
  • Fusion strategy: low rank
  • Task head: retrieval
  • Activation: gelu
  • Normalization: scalenorm
  • Initialization: orthogonal

Training

  • Optimizer: adam
  • LR scheduler: linear warmup

Files

  • model_112034448_flamingo_nano.py — main artifact of this repository

License

See the license field above.