Matthewmiller59's picture
Upload 2 files
cd1a025 verified
|
Raw
History Blame Contribute Delete
754 Bytes
metadata
license: cc-by-4.0
tags:
  - deit
  - generation
  - instancenorm
  - kaiming-normal
  - linear
  - low-rank
  - onecycle
  - sgd
  - swish
  - tiny

model_563159428_deit_tiny.py

Model Overview

A tiny-scale implementation of the deit architecture, built for generation tasks.

Architecture

  • Architecture: deit
  • Scale: tiny
  • Attention: linear
  • Fusion strategy: low rank
  • Task head: generation
  • Activation: swish
  • Normalization: instancenorm
  • Initialization: kaiming normal

Training

  • Optimizer: sgd
  • LR scheduler: onecycle

Files

  • model_563159428_deit_tiny.py — main artifact of this repository

License

See the license field above.