fix quant unit test (#25792)

5 years ago · f6ac5990aa
parent 2131559d08
commit f6ac5990aa
1 changed files with 1 additions and 0 deletions
--- a/python/paddle/fluid/contrib/slim/tests/test_user_defined_quantization.py
+++ b/python/paddle/fluid/contrib/slim/tests/test_user_defined_quantization.py
@ -47,6 +47,7 @@ def residual_block(img, label, num=1):
            num_filters=ch_out,
            stride=stride,
            padding=padding,
+            use_cudnn=False,
            act=None,
            bias_attr=bias_attr)
        return fluid.layers.batch_norm(input=tmp, act=act)