From c01618d7b09c8aa1b7e7aa2c4785cbf04031cb3c Mon Sep 17 00:00:00 2001 From: ZihengJiang Date: Mon, 12 Aug 2019 14:28:38 -0700 Subject: [PATCH] Update test. --- .../python/nightly/quantization/test_quantization_accuracy.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tests/python/nightly/quantization/test_quantization_accuracy.py b/tests/python/nightly/quantization/test_quantization_accuracy.py index 6f422e37ce659..eb7a1717b6a55 100644 --- a/tests/python/nightly/quantization/test_quantization_accuracy.py +++ b/tests/python/nightly/quantization/test_quantization_accuracy.py @@ -134,6 +134,7 @@ def test_quantize_acc(cfg): results = [] configs = [ Config('mobilenetv2_1.0', nbit_input=8, dtype_input='int8', nbit_output=32, dtype_output='int32', global_scale=4.0, expected_acc=0.666), + Config('resnet18_v1', nbit_input=8, dtype_input='int8', nbit_output=16, dtype_output='int16', global_scale=8.0, expected_acc=0.692), Config('resnet18_v1', nbit_input=8, dtype_input='int8', nbit_output=32, dtype_output='int32', global_scale=8.0, expected_acc=0.692), Config('resnet34_v1', nbit_input=8, dtype_input='int8', nbit_output=32, dtype_output='int32', global_scale=8.0, expected_acc=0.733), @@ -146,4 +147,5 @@ def test_quantize_acc(cfg): for config in configs: acc = test_quantize_acc(config) results.append((config, acc)) - print(results) + for res in results: + print(res)