gokulsrinivasagan/processed_wikitext-103-raw-v1-ld
Viewer โข Updated โข 230k โข 10
How to use gokulsrinivasagan/distilbert_lda_100 with Transformers:
# pip install -U transformers accelerate
# Load model directly
from transformers import AutoTokenizer, DistilBertForLDAMaskedLM
tokenizer = AutoTokenizer.from_pretrained("gokulsrinivasagan/distilbert_lda_100")
model = DistilBertForLDAMaskedLM.from_pretrained("gokulsrinivasagan/distilbert_lda_100", 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.3242 | 4.1982 | 10000 | 8.2423 | 0.1513 |
| 6.3124 | 8.3963 | 20000 | 6.0353 | 0.3573 |
| 4.9214 | 12.5945 | 30000 | 4.7812 | 0.5051 |
| 4.5891 | 16.7926 | 40000 | 4.4732 | 0.5490 |
| 4.4469 | 20.9908 | 50000 | 4.3617 | 0.5604 |