gokulsrinivasagan/processed_wikitext-103-raw-v1-ld
Viewer • Updated • 230k • 13
How to use gokulsrinivasagan/tinybert_base_train_book_ent_15p_lda with Transformers:
# Load model directly
from transformers import AutoTokenizer, DistilBertForLDAMaskedLM
tokenizer = AutoTokenizer.from_pretrained("gokulsrinivasagan/tinybert_base_train_book_ent_15p_lda")
model = DistilBertForLDAMaskedLM.from_pretrained("gokulsrinivasagan/tinybert_base_train_book_ent_15p_lda", 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.0365 | 8.7413 | 10000 | 7.7884 | 0.2552 |
| 6.6825 | 17.4825 | 20000 | 6.6636 | 0.3388 |