model_656390823_mae_xlarge.py

Model Overview

A xlarge-scale implementation of the mae architecture, built for matching tasks.

Architecture

• Architecture: mae
• Scale: xlarge
• Attention: sparse
• Fusion strategy: tucker
• Task head: matching
• Activation: gelu
• Normalization: layernorm
• Initialization: xavier uniform

Training

• Optimizer: lamb
• LR scheduler: polynomial

Files

• model_656390823_mae_xlarge.py — main artifact of this repository

License

See the license field above.