samsontsai's picture
Upload 2 files
a70f14b verified
|
Raw
History Blame Contribute Delete
760 Bytes
metadata
license: mit
tags:
  - beit
  - co-attention
  - flash
  - groupnorm
  - huge
  - linear-warmup
  - matching
  - rmsprop
  - swish
  - xavier-uniform

model_624243866_beit_huge.py

Model Overview

A huge-scale implementation of the beit architecture, built for matching tasks.

Architecture

  • Architecture: beit
  • Scale: huge
  • Attention: flash
  • Fusion strategy: co attention
  • Task head: matching
  • Activation: swish
  • Normalization: groupnorm
  • Initialization: xavier uniform

Training

  • Optimizer: rmsprop
  • LR scheduler: linear warmup

Files

  • model_624243866_beit_huge.py — main artifact of this repository

License

See the license field above.