jacksoncourtney
/

bart-translate

tiny-transformer

Model card Files Files and versions

predict.py

Model Overview

A nano-scale implementation of the tiny transformer architecture, built for multitask tasks.

Architecture

Architecture: tiny transformer
Scale: nano
Attention: multi query
Fusion strategy: tensor fusion
Task head: multitask
Activation: relu
Normalization: scalenorm
Initialization: xavier

Training

Optimizer: rmsprop
LR scheduler: polynomial

Files

predict.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