diff --git a/examples/bert/bert_inc_ptq_cpu.json b/examples/bert/bert_inc_ptq_cpu.json index dde9407fb..180b2ad38 100644 --- 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 index 12ab18944..0df1ff8b5 100644 --- 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",