Skip to content

Pre-trained Models v1.0

Latest

Choose a tag to compare

@uditrana uditrana released this 29 Oct 05:29
· 23 commits to main since this release

Pre-trained best models from the paper. Includes:

  • Zoo Architecture (M128_L2): 888K checkpoint
  • Transformer (M512_L2): 77M checkpoint
  • Test set predictions for both models

Usage:

# Download and extract models
uv run dvc repro prep_data download_pretrained_models generate_results_fasttrack

This skips the expensive training stages and downloads pre-trained models directly.