提交 c020f5cd 编写于 作者: W wanghaoshuang

Fix unitest of quant.

上级 a596e2b6
...@@ -77,9 +77,8 @@ class TestQuantPostOnlyWeightCase1(unittest.TestCase): ...@@ -77,9 +77,8 @@ class TestQuantPostOnlyWeightCase1(unittest.TestCase):
fetch_list=outputs) fetch_list=outputs)
iter += 1 iter += 1
if iter % 100 == 0: if iter % 100 == 0:
print( print('eval iter={}, avg loss {}, acc_top1 {}, acc_top5 {}'.
'eval iter={}, avg loss {}, acc_top1 {}, acc_top5 {}'. format(iter, cost, top1, top5))
format(iter, cost, top1, top5))
result[0].append(cost) result[0].append(cost)
result[1].append(top1) result[1].append(top1)
result[2].append(top5) result[2].append(top5)
...@@ -99,7 +98,7 @@ class TestQuantPostOnlyWeightCase1(unittest.TestCase): ...@@ -99,7 +98,7 @@ class TestQuantPostOnlyWeightCase1(unittest.TestCase):
params_filename='params') params_filename='params')
quant_post_dynamic( quant_post_dynamic(
model_dir='./test_quant_post', model_dir='./test_quant_post_dynamic',
save_model_dir='./test_quant_post_inference', save_model_dir='./test_quant_post_inference',
model_filename='model', model_filename='model',
params_filename='params', params_filename='params',
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册