You can deploy this model and start earning money today!

bansalaman18/
bert-uncased_L-2_H-128_A-2

Name: bert-uncased_L-2_H-128_A-2
Availability: InStock

Views

No views yet

BERT bert-uncased_L-2_H-128_A-2

This model is a PyTorch conversion of the original TensorFlow BERT checkpoint.

Model Details

Model Type: BERT (Bidirectional Encoder Representations from Transformers)
Language: English (uncased)
Architecture:
- Layers: 2
- Hidden Size: 128
- Attention Heads: 2
- Vocabulary Size: 30522
- Max Position Embeddings: 512

Model Configuration

json

{
  "hidden_size": 128,
  "hidden_act": "gelu",
  "initializer_range": 0.02,
  "vocab_size": 30522,
  "hidden_dropout_prob": 0.1,
  "num_attention_heads": 2,
  "type_vocab_size": 2,
  "max_position_embeddings": 512,
  "num_hidden_layers": 2,
  "intermediate_size": 512,
  "attention_probs_dropout_prob": 0.1
}

Usage

python

from transformers import BertForPreTraining, BertTokenizer

# Load the model and tokenizer
model = BertForPreTraining.from_pretrained('bansalaman18/bert-uncased_L-2_H-128_A-2')
tokenizer = BertTokenizer.from_pretrained('bansalaman18/bert-uncased_L-2_H-128_A-2')

# Example usage
text = "Hello, this is a sample text for BERT."
inputs = tokenizer(text, return_tensors='pt')
outputs = model(**inputs)

Training Data

This model was originally trained on the same data as the standard BERT models:

English Wikipedia (2500M words)
BookCorpus (800M words)

Conversion Details

This model was converted from the original TensorFlow checkpoint to PyTorch format using a custom conversion script with the Hugging Face Transformers library.

Citation

bibtex

@article{devlin2018bert,
  title={BERT: Pre-training of Deep Bidirectional Transformers for Language Understanding},
  author={Devlin, Jacob and Chang, Ming-Wei and Lee, Kenton and Toutanova, Kristina},
  journal={arXiv preprint arXiv:1810.04805},
  year={2018}
}