← all repositories

lonePatient/albert_pytorch

A PyTorch implementation of ALBERT, a parameter-efficient variant of BERT for self-supervised language understanding.

715 stars Python Language ModelsML Frameworks
albert_pytorch
Velocity · 7d
+0.3
★ / day
Trend
steady
star history

This repository provides a PyTorch implementation of ALBERT (A Lite BERT), a parameter-efficient transformer model designed for self-supervised learning of language representations. It includes pre-trained model downloads, tools to convert TensorFlow checkpoints to PyTorch format, and fine-tuning scripts for downstream NLP tasks. The implementation supports multiple model sizes (base, large, xlarge, xxlarge) and both v1 and v2 releases.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.