pytorch tf rust distilbert text-classification en dataset:sst-2 transformers license:apache-2.0
.gitattributes README.md config.json pytorch_model.bin rust_model.ot tf_model.h5 tokenizer_config.json vocab.txt
---
language: en
license: apache-2.0
datasets:
- sst-2
---

# DistilBERT base uncased finetuned SST-2

This model is a fine-tune checkpoint of [DistilBERT-base-uncased](https://huggingface.co/distilbert-base-uncased), fine-tuned on SST-2.
This model reaches an accuracy of 91.3 on the dev set (for comparison, Bert bert-base-uncased version reaches an accuracy of 92.7).

# Fine-tuning hyper-parameters

- learning_rate = 1e-5
- batch_size = 32
- warmup = 600
- max_seq_length = 128
- num_train_epochs = 3.0
