From c020f5cd306440b63b0c409ad6f403336770bce5 Mon Sep 17 00:00:00 2001
From: wanghaoshuang <wanghaoshuang@baidu.com>
Date: Tue, 14 Jul 2020 08:47:34 +0000
Subject: [PATCH] Fix unitest of quant.

---
 tests/test_quant_post_only_weight.py | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/tests/test_quant_post_only_weight.py b/tests/test_quant_post_only_weight.py
index e90b13e6..ede4094d 100644
--- a/tests/test_quant_post_only_weight.py
+++ b/tests/test_quant_post_only_weight.py
@@ -77,9 +77,8 @@ class TestQuantPostOnlyWeightCase1(unittest.TestCase):
                                            fetch_list=outputs)
                 iter += 1
                 if iter % 100 == 0:
-                    print(
-                        'eval iter={}, avg loss {}, acc_top1 {}, acc_top5 {}'.
-                        format(iter, cost, top1, top5))
+                    print('eval iter={}, avg loss {}, acc_top1 {}, acc_top5 {}'.
+                          format(iter, cost, top1, top5))
                 result[0].append(cost)
                 result[1].append(top1)
                 result[2].append(top5)
@@ -99,7 +98,7 @@ class TestQuantPostOnlyWeightCase1(unittest.TestCase):
             params_filename='params')
 
         quant_post_dynamic(
-            model_dir='./test_quant_post',
+            model_dir='./test_quant_post_dynamic',
             save_model_dir='./test_quant_post_inference',
             model_filename='model',
             params_filename='params',
-- 
GitLab