gokulsrinivasagan/processed_wikitext-103-raw-v1-ld
Viewer • Updated • 230k • 15
How to use gokulsrinivasagan/distilbert_lda_50 with Transformers:
# Load model directly
from transformers import AutoTokenizer, DistilBertForLDAMaskedLM
tokenizer = AutoTokenizer.from_pretrained("gokulsrinivasagan/distilbert_lda_50")
model = DistilBertForLDAMaskedLM.from_pretrained("gokulsrinivasagan/distilbert_lda_50", 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.325 | 4.1982 | 10000 | 8.2433 | 0.1515 |
| 6.4392 | 8.3963 | 20000 | 6.1265 | 0.3497 |
| 4.9408 | 12.5945 | 30000 | 4.7936 | 0.5043 |
| 4.6033 | 16.7926 | 40000 | 4.4840 | 0.5467 |
| 4.453 | 20.9908 | 50000 | 4.3695 | 0.5585 |