aademir7's picture
Upload 2 files
5dd9958 verified
|
Raw
History Blame Contribute Delete
764 Bytes
metadata
license: cc-by-4.0
tags:
  - adam
  - albef
  - batchnorm
  - exponential
  - gated-fusion
  - relu
  - retrieval
  - standard
  - tiny
  - trunc-normal

model_227866586_albef_tiny.py

Model Overview

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

Architecture

  • Architecture: albef
  • Scale: tiny
  • Attention: standard
  • Fusion strategy: gated fusion
  • Task head: retrieval
  • Activation: relu
  • Normalization: batchnorm
  • Initialization: trunc normal

Training

  • Optimizer: adam
  • LR scheduler: exponential

Files

  • model_227866586_albef_tiny.py — main artifact of this repository

License

See the license field above.