gokulsrinivasagan/processed_wikitext-103-raw-v1-ld
Viewer • Updated • 230k • 11
How to use gokulsrinivasagan/distilbert_lda_20 with Transformers:
# pip install -U transformers accelerate
# Load model directly
from transformers import AutoTokenizer, DistilBertForLDAMaskedLM
tokenizer = AutoTokenizer.from_pretrained("gokulsrinivasagan/distilbert_lda_20")
model = DistilBertForLDAMaskedLM.from_pretrained("gokulsrinivasagan/distilbert_lda_20", device_map="auto")This model is a fine-tuned version of on the gokulsrinivasagan/processed_wikitext-103-raw-v1-ld dataset. It achieves the following results on the evaluation set:
More information needed
More information needed
More information needed
The following hyperparameters were used during training:
| Training Loss | Epoch | Step | Validation Loss | Accuracy |
|---|---|---|---|---|
| 8.3243 | 4.1982 | 10000 | 8.2446 | 0.1514 |
| 5.789 | 8.3963 | 20000 | 5.5968 | 0.4033 |
| 4.7261 | 12.5945 | 30000 | 4.6120 | 0.5311 |
| 4.4663 | 16.7926 | 40000 | 4.3618 | 0.5648 |
| 4.3476 | 20.9908 | 50000 | 4.2686 | 0.5743 |