20 lines
508 B
Markdown
20 lines
508 B
Markdown
|
---
|
||
|
|
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
|