← all repositories

google-research/electra

ELECTRA is a self-supervised method for pre-training transformer text encoders using a discriminator that distinguishes real tokens from generated fakes.

2.4k stars Python Language ModelsML Frameworks
electra
Velocity · 7d
+1.0
★ / day
Trend
steady
star history

The repository provides code to pre-train ELECTRA models, including small models trainable on a single GPU, and supports fine-tuning on downstream NLP tasks including GLUE classification, SQuAD question answering, and sequence tagging. It also includes Electric, an energy-based version of ELECTRA that can produce pseudo-likelihood scores for re-ranking speech or translation outputs.

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