CoolFace
Modelpublic

dvyomkesh/nemo-aot-o3-full-unsloth-r32-1xrtxpro6000

sourceHugging Faceotherupdated 4mo agoView on Hugging Face
0likes
Model Card

Nemo AoT-O3 Full Unsloth SFT LoRA r32 (0.62 — Regression)

Kaggle Public Score: 0.62 — regression from the 0.83 baseline. Kept as an ablation artifact.

Regression Analysis

This adapter scored 0.62 (vs 0.83 baseline) due to several compounding issues:

FactorThis run0.83 baseline
Starting pointFrom scratch (no initial_adapter)N/A (was the first adapter)
Learning rate2e-4, 0 warmup2e-4, 0 warmup
Gradient clipping1e9 (none)1e9 (none)
SamplingDisabled (flat 7,524 rows)N/A (different dataset)
DatasetAoT-O3 canonical (7,524 rows)DGXChen/Tong original
Initial loss67.98 (uncalibrated LoRA)~1.5 (pretrained base)

Root cause: Training a new LoRA from scratch on the AoT-O3 dataset with aggressive hyperparameters (lr=2e-4, no warmup, no grad clipping) caused the first ~10 steps to have catastrophic gradient updates (loss 67→5, gradnorm up to 17.9). The model recovered to lossmean_20=2.47 but the early corruption was unrecoverable.

Lesson: When training on the AoT-O3 dataset, use the 0.83 adapter as initial_adapter (continuation SFT) with conservative hyperparameters (lr=2e-5, warmup=20, maxgradnorm=1.0).

Training Configuration

ParameterValue
Base modelnvidia/NVIDIA-Nemotron-3-Nano-30B-A3B-BF16
BackendUnsloth (4-bit quantized base)
LoRA rank32
LoRA alpha32
Target modulesq_proj, k_proj, v_proj, o_proj, in_proj, out_proj, up_proj, down_proj
Max sequence length8,192
Epochs1 (236 steps)
Global effective batch32 (1 × 1 GPU × 32 grad accum)
Learning rate2e-4 (linear decay, 0 warmup)
Optimizeradamw_8bit
Gradient clipping1e9 (effectively none)
Hardware1× RTX PRO 6000 (Blackwell)
Training time371.8 min
Final lossmean202.47
Epoch mean loss7.29
W&Blitefold-research/nemo-nemotron-challenge/ynwngeee

Dataset

`dvyomkesh/nemo-aot-o3-style-reasoning` — 7,524 rows, all gate-pass, assistant-only labels.

Sampling was disabled — no oversampling of scarce types (cryptarithm_guess: 164, equation_numeric_guess: 136).

Kaggle Submission

  • —Ref: 52803700
  • —Public Score: 0.62
  • —Status: COMPLETE

Contents

adapter/                  # PEFT LoRA checkpoint + tokenizer
submission/submission.zip # Exact Kaggle submission archive
repro/                    # Config and scripts for reproduction
logs/                     # Modal training log, Kaggle submission logs, W&B URL
metadata/                 # Manifest, checksums, label mask summary, sampling summary

Do Not Use This Adapter For

  • —Continuation training — the weights are corrupted by the early high-loss phase
  • —Submission — it regresses from the 0.83 baseline
  • —Benchmarking the AoT-O3 dataset — the regression is due to hyperparameters, not data quality