stanfordnlp/imdb
Viewer • Updated • 100k • 170k • 1.27k
How to use fengdavid/my_awesome_model with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("image-classification", model="fengdavid/my_awesome_model")
pipe("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/hub/parrots.png") # pip install -U transformers accelerate
# Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("fengdavid/my_awesome_model")
model = AutoModelForSequenceClassification.from_pretrained("fengdavid/my_awesome_model", device_map="auto")This model is a fine-tuned version of distbert on the blala 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 |
|---|---|---|---|---|
| No log | 1.0 | 391 | 0.1959 | 0.9243 |
| 0.2489 | 2.0 | 782 | 0.1908 | 0.929 |