xinhe commited on
Commit
25121a8
1 Parent(s): ce448ed

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +31 -0
README.md CHANGED
@@ -1,3 +1,34 @@
1
  ---
2
  license: apache-2.0
 
 
 
 
 
 
 
 
 
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: apache-2.0
3
+ tags:
4
+ - int8
5
+ - Intel® Neural Compressor
6
+ - neural-compressor
7
+ - PostTrainingDynamic
8
+ datasets:
9
+ - cnn_dailymail
10
+ metrics:
11
+ - rougeLsum
12
  ---
13
+
14
+ # INT8 DistilBart finetuned on CNN DailyMail
15
+ ### Post-training dynamic quantization
16
+ This is an INT8 PyTorch model quantized with [huggingface/optimum-intel](https://github.com/huggingface/optimum-intel) through the usage of [Intel® Neural Compressor](https://github.com/intel/neural-compressor).
17
+ The original fp32 model comes from the fine-tuned model [sysresearch101/t5-large-finetuned-xsum-cnn](https://huggingface.co/sysresearch101/t5-large-finetuned-xsum-cnn).
18
+
19
+ Below linear modules are fallbacked to fp32 for less than 1% relative accuracy loss:
20
+
21
+ **'model.decoder.layers.2.fc2'**, **'model.encoder.layers.11.fc2'**, **'model.decoder.layers.1.fc2'**, **'model.decoder.layers.0.fc2'**, **'model.decoder.layers.4.fc1'**, **'model.decoder.layers.3.fc2'**, **'model.encoder.layers.8.fc2'**, **'model.decoder.layers.3.fc1'**, **'model.encoder.layers.11.fc1'**, **'model.encoder.layers.0.fc2'**, **'model.encoder.layers.3.fc1'**, **'model.encoder.layers.10.fc2'**, **'model.decoder.layers.5.fc1'**, **'model.encoder.layers.1.fc2'**, **'model.encoder.layers.3.fc2'**, **'lm_head'**, **'model.encoder.layers.7.fc2'**, **'model.decoder.layers.0.fc1'**, **'model.encoder.layers.4.fc1'**, **'model.encoder.layers.10.fc1'**, **'model.encoder.layers.6.fc1'**
22
+
23
+ ### Evaluation result
24
+ | |INT8|FP32|
25
+ |---|:---:|:---:|
26
+ | **Accuracy (eval-rougeLsum)** | 41.4707 | 41.8117 |
27
+ | **Model size** |722M|1249M|
28
+ ### Load with optimum:
29
+ ```python
30
+ from optimum.intel.neural_compressor.quantization import IncQuantizedModelForSeq2SeqLM
31
+ int8_model = IncQuantizedModelForSeq2SeqLM.from_pretrained(
32
+ 'Intel/distilbart-cnn-12-6-int8-dynamic',
33
+ )
34
+ ```