README.md

metadata

language:
  - en
license: apache-2.0
tags:
  - text-classfication
  - int8
  - Intel® Neural Compressor
  - PostTrainingDynamic
datasets:
  - glue
metrics:
  - f1
model-index:
  - name: bart-large-mrpc-int8-dynamic
    results:
      - task:
          name: Text Classification
          type: text-classification
        dataset:
          name: GLUE MRPC
          type: glue
          args: mrpc
        metrics:
          - name: F1
            type: f1
            value: 0.9050847457627118

INT8 bart-large-mrpc

Post-training dynamic quantization

This is an INT8 PyTorch model quantized with huggingface/optimum-intel through the usage of Intel® Neural Compressor.

The original fp32 model comes from the fine-tuned model bart-large-mrpc.

Test result

	INT8	FP32
Accuracy (eval-f1)	0.9051	0.9120
Model size (MB)	547	1556.48

Load with optimum:

from optimum.intel.neural_compressor.quantization import IncQuantizedModelForSequenceClassification
int8_model = IncQuantizedModelForSequenceClassification.from_pretrained(
    'Intel/bart-large-mrpc-int8-dynamic',
)