Downloads · 30 days
24
6% of all-time downloads
distilbert-amazon-shoe-reviews-onnx-optimized
juliensimon/distilbert-amazon-shoe-reviews-onnx-optimized
distilbert-amazon-shoe-reviews-onnx-optimized is a text classification model from juliensimon. Use it when you need a label for a piece of text. It is set up for transformers.
Graph-optimized ONNX export of juliensimon/distilbert-amazon-shoe-reviews for faster CPU inference using Optimum.
All-time downloads
433
Public
Repo size
536 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.onnx268 MB · 100%
At a glance
- Task
- Text Classification
- Library
- transformers
- Access
- Public
- Created
- Jun 9, 2022
- Updated
- Mar 22, 2026
- SHA
- acae2d3b
Try a prompt
Base models
Model card
From the Hugging Face model README
DistilBERT Amazon Shoe Reviews — ONNX Optimized
Graph-optimized ONNX export of juliensimon/distilbert-amazon-shoe-reviews for faster CPU inference using Optimum.
Video walkthrough: Accelerate Transformer inference on CPU with Optimum and ONNX
Model Details
| Detail | Value |
|---|---|
| Base model | juliensimon/distilbert-amazon-shoe-reviews |
| Task | Star rating prediction (1–5 stars) from shoe reviews |
| Format | ONNX with graph optimizations (operator fusion, constant folding) |
Usage
from optimum.onnxruntime import ORTModelForSequenceClassification
from transformers import AutoTokenizer, pipeline
model = ORTModelForSequenceClassification.from_pretrained("juliensimon/distilbert-amazon-shoe-reviews-onnx-optimized")
tokenizer = AutoTokenizer.from_pretrained("juliensimon/distilbert-amazon-shoe-reviews-onnx-optimized")
classifier = pipeline("text-classification", model=model, tokenizer=tokenizer)
classifier("These shoes are incredibly comfortable!")
Related Models
- distilbert-amazon-shoe-reviews — Original PyTorch model
- distilbert-amazon-shoe-reviews-onnx — Vanilla ONNX export
- distilbert-amazon-shoe-reviews-onnx-quantized — ONNX with dynamic quantization
- distilbert-amazon-shoe-reviews-quantized — PyTorch dynamic quantization
- Task
- Text Classification
- Library
- transformers
- Created
- Jun 9, 2022
- Updated
- Mar 22, 2026