CoolFace
Modelpublic

deepvk/bert-base-uncased

sourceHugging Faceapache-2.0updated 3y agoView on Hugging Face
5likes112downloads
Model Card

BERT-base

<!-- Provide a quick summary of what the model is/does. -->

Pretrained bidirectional encoder for russian language. The model was trained using standard MLM objective on large text corpora including open social data. See Training Details section for more information.

⚠️ This model contains only the encoder part without any pretrained head.

  • —Developed by: deepvk
  • —Model type: BERT
  • —Languages: Mostly russian and small fraction of other languages
  • —License: Apache 2.0

How to Get Started with the Model

python
from transformers import AutoTokenizer, AutoModel

tokenizer = AutoTokenizer.from_pretrained("deepvk/bert-base-uncased")
model = AutoModel.from_pretrained("deepvk/bert-base-uncased")

text = "Привет, мир!"

inputs = tokenizer(text, return_tensors='pt')
predictions = model(**inputs)

Training Details

The model was trained using the NVIDIA source code. See the pretraining documentation for details.

Training Data

250 GB of filtered texts in total. A mix of the following data: Wikipedia, Books and Social corpus.

Architecture details

ArgumentValue
Encoder layers12
Encoder attention heads12
Encoder embed dim768
Encoder ffn embed dim3,072
Activation functionGeLU
Attention dropout0.1
Dropout0.1
Max positions512
Vocab size36000
Tokenizer typeBertTokenizer

Evaluation

We evaluated the model on Russian Super Glue dev set. The best result in each task is marked in bold. All models have the same size except the distilled version of DeBERTa.

ModelRCBPARusMuSeRCTERRaRUSSERWSDDaNetQAScore
vk-deberta-distill0.4330.560.6250.590.9430.5690.7260.635
vk-roberta-base0.460.560.6790.7690.9600.5690.6580.665
vk-deberta-base0.4500.610.7220.7040.9480.5780.760.682
vk-bert-base0.4670.570.5870.7040.9530.5830.7370.657
sber-bert-base0.4910.610.6630.7690.9620.5740.6780.678