fix config of bert inc example

Signed-off-by: yuwenzho <[email protected]>
microsoft · Jul 13, 2023 · 1619952 · 1619952
1 parent 1035fb6
commit 1619952
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/examples/bert/bert_inc_ptq_cpu.json b/examples/bert/bert_inc_ptq_cpu.json
@@ -56,6 +56,7 @@
         },
         "quantization": {
             "type": "IncQuantization",
+            "disable_search": true,
             "config": {
                 "approach": "SEARCHABLE_VALUES",
                 "user_script": "user_script.py",

diff --git a/examples/bert/bert_inc_smoothquant_ptq_cpu.json b/examples/bert/bert_inc_smoothquant_ptq_cpu.json
@@ -17,13 +17,14 @@
             "metrics":[
                 {
                     "name": "accuracy",
-                    "type": "custom",
+                    "type": "accuracy",
                     "sub_types": [
-                        {"name": "accuracy_custom", "priority": 1, "higher_is_better": true, "goal": {"type": "percent-max-degradation", "value": 2}}
+                        {"name": "accuracy_score", "priority": 1, "goal": {"type": "percent-max-degradation", "value": 2}}
                     ],
                     "user_config":{
+                        "post_processing_func": "post_process",
                         "user_script": "user_script.py",
-                        "evaluate_func": "eval_accuracy",
+                        "dataloader_func": "create_dataloader",
                         "batch_size": 1
                     }
                 },
@@ -62,7 +63,7 @@
                 "dataloader_func": "inc_glue_calibration_reader",
                 "recipes":{
                     "smooth_quant": true,
-                    "smooth_quant_args": {"alpha": 0.6}
+                    "smooth_quant_args": {"alpha": 0.7}
                 },
                 "metric": {
                     "name": "accuracy",