How to Get Started With the Model

PyTorch

To load the quantized model, you can do as follows:

python

from optimum.intel import INCModelForSequenceClassification

model_id = "distilbert-base-uncased-finetuned-sst-2-english-int8-dynamic-inc"
model = INCModelForSequenceClassification.from_pretrained(model_id)

ONNX

This is an INT8 ONNX model quantized with Intel® Neural Compressor.

The original fp32 model comes from the fine-tuned model DistilBERT.

Test result

	INT8	FP32
Accuracy (eval-accuracy)	0.9025	0.9106
Model size (MB)	165	256

Load ONNX model:

python

from optimum.onnxruntime import ORTModelForSequenceClassification
model = ORTModelForSequenceClassification.from_pretrained('Intel/distilbert-base-uncased-finetuned-sst-2-english-int8-dynamic')

Intel/distilbert-base-uncased-finetuned-sst-2-english-int8-dynamic-inc

Dynamically quantized DistilBERT base uncased finetuned SST-2

Table of Contents

Model Details

How to Get Started With the Model

PyTorch

ONNX

Test result

Load ONNX model:

Intel/
distilbert-base-uncased-finetuned-sst-2-english-int8-dynamic-inc