Downloads · 30 days
6
32% of all-time downloads
WattsIshaan/OLMo-60m-192B-adamw-wsd-10percent
OLMo-60m-192B-adamw-wsd-10percent 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-60M checkpoint pretrained with AdamW optimizer and WSD LR schedule on 120B tokens from DCLM. This checkpoint is for 90% of the training steps (670000 steps at 256 bsz), and can be annealed for the remaining 10% s…
Downloads · 30 days
6
32% of all-time downloads
All-time downloads
19
Public
Repo size
1.6 GB
Likes
1
Public
Click a slice to open those files.
.pt1.6 GB · 100%
From the Hugging Face model README
OLMo-60M checkpoint pretrained with AdamW optimizer and WSD LR schedule on 120B tokens from DCLM. This checkpoint is for 90% of the training steps (670000 steps at 256 bsz), and can be annealed for the remaining 10% steps.
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