Downloads · 30 days
7
32% of all-time downloads
WattsIshaan/OLMo-150m-120B-sam-cosine
OLMo-150m-120B-sam-cosine is a machine learning model from WattsIshaan. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as cc-by-4.0.
OLMo-150M checkpoint pretrained with SAM optimizer (rho 5e-2) and cosine LR schedule on 120B tokens from DCLM.
Downloads · 30 days
7
32% of all-time downloads
All-time downloads
22
Public
Repo size
3.2 GB
Likes
0
Public
Click a slice to open those files.
.pt3.2 GB · 100%
From the Hugging Face model README
OLMo-150M checkpoint pretrained with SAM optimizer (rho 5e-2) and cosine LR schedule on 120B tokens from DCLM.
Paper:"Sharpness-Aware Pretraining Mitigates Catastrophic Forgetting" Ishaan Watts*, Catherine Li*, Sachin Goyal, Jacob Mitchell Springer, Aditi Raghunathan ICML2026 https://arxiv.org/abs/2605.02105