finetune.py

Model Overview

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

Architecture

  • Architecture: deit
  • Scale: huge
  • Attention: standard
  • Fusion strategy: concat mlp
  • Task head: generation
  • Activation: relu
  • Normalization: instancenorm
  • Initialization: kaiming normal

Training

  • Optimizer: sgd
  • LR scheduler: onecycle

Files

  • finetune.py — main artifact of this repository

License

See the license field above.

Downloads last month

-

Downloads are not tracked for this model. How to track
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support