Instructions to use VinayHajare/quickdraw-mobilevit-small with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use VinayHajare/quickdraw-mobilevit-small with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("image-classification", model="VinayHajare/quickdraw-mobilevit-small") pipe("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/hub/parrots.png")# Load model directly from transformers import AutoImageProcessor, AutoModelForImageClassification processor = AutoImageProcessor.from_pretrained("VinayHajare/quickdraw-mobilevit-small") model = AutoModelForImageClassification.from_pretrained("VinayHajare/quickdraw-mobilevit-small", device_map="auto") - Notebooks
- Google Colab
- Kaggle
File size: 380 Bytes
2f11503 | 1 2 3 4 5 6 7 8 9 10 11 12 13 | {
"epoch": 5.0,
"eval_accuracy": 0.751164,
"eval_loss": 0.9863214492797852,
"eval_runtime": 104.5218,
"eval_samples_per_second": 2391.845,
"eval_steps_per_second": 9.347,
"total_flos": 5.4597447576e+17,
"train_loss": 1.1272559640920097,
"train_runtime": 10316.2205,
"train_samples_per_second": 2181.031,
"train_steps_per_second": 8.52
} |