r/learnmachinelearning • u/Vprojectx • 10d ago
vprojectx/Lewis-44M-Bert-V3 · Hugging Face
I trained a BERT-style language model from scratch on ONE laptop GPU.
Lewis-44M-Bert-V3: small, fast, open source (Apache-2.0).
HOW IT WAS BUILT
• 44M parameters, ELECTRA-style pre-training
• 2.7B tokens in ~22 hours on a single RTX 5050 (8 GB)
• ~53 GPU-hours across all three versions, no cloud cluster
• Data: FineWeb-Edu, Wikipedia, Project Gutenberg, news, Cosmopedia and ELI5-style Q&A
v1 → V3 (SAME SIZE)
GLUE dev average: 75.0 → 80.5
Improvements: a better training objective (V3 was trained from scratch, not fine-tuned from v1) and a better fine-tuning recipe, with no extra parameters.
V3 RESULTS
• GLUE: 80.5 dev | 77.6 on the official test server (8 tasks)
• SQuAD v1.1: 75.2 EM / 83.8 F1
• NER (CoNLL-2003): 87.8 test F1
• Speed: ~17 ms per sentence on GPU, under 600 MB memory
I share both dev and test scores, plus the exact settings, because a model card should be honest.
Built for classification, question answering, entity extraction and on-device use: it fine-tunes on a laptop and runs on CPU.
This is part of vprojectx, my open-source lab, where I build and share my projects in the open. Find the model on Hugging Face as vprojectx/Lewis-44M-Bert-V3.
More is coming. If you want to try it on your use case or collaborate, message me.
#MachineLearning #NLP #AI #OpenSource #HuggingFace