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 f1856b18be462dc537e6029cf8394757118008ad..954ed2945fd9693a6d567246461ad9db40ebcfb6 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)