ajaysha0723's picture
Upload 2 files
2fd788d verified
|
Raw
History Blame Contribute Delete
777 Bytes
metadata
license: cc-by-4.0
tags:
  - adafactor
  - constant-warmup
  - gated-fusion
  - generation
  - instancenorm
  - kaiming
  - mae
  - multi-query
  - nano
  - relu

model_634861505_mae_nano.py

Model Overview

A nano-scale implementation of the mae architecture, built for generation tasks.

Architecture

  • Architecture: mae
  • Scale: nano
  • Attention: multi query
  • Fusion strategy: gated fusion
  • Task head: generation
  • Activation: relu
  • Normalization: instancenorm
  • Initialization: kaiming

Training

  • Optimizer: adafactor
  • LR scheduler: constant warmup

Files

  • model_634861505_mae_nano.py — main artifact of this repository

License

See the license field above.