From 7dbc5126ba11daabcabe1411db7f2ded12e53247 Mon Sep 17 00:00:00 2001 From: Bai Yifan Date: Thu, 17 Sep 2020 19:49:03 +0800 Subject: [PATCH] fix user_defined quant unittest (#27385) --- .../fluid/contrib/slim/tests/test_user_defined_quantization.py | 1 + 1 file changed, 1 insertion(+) diff --git a/python/paddle/fluid/contrib/slim/tests/test_user_defined_quantization.py b/python/paddle/fluid/contrib/slim/tests/test_user_defined_quantization.py index f1856b18be4..954ed2945fd 100644 --- a/python/paddle/fluid/contrib/slim/tests/test_user_defined_quantization.py +++ b/python/paddle/fluid/contrib/slim/tests/test_user_defined_quantization.py @@ -48,6 +48,7 @@ def residual_block(img, label, num=1): num_filters=ch_out, stride=stride, padding=padding, + use_cudnn=use_cudnn, act=None, bias_attr=bias_attr) return fluid.layers.batch_norm(input=tmp, act=act) -- GitLab