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 0c8f5cdd84cd486899cb892999b103cab72bfebb..c9ea15bf6cde9af16810920f53a7d5e045a852e3 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 @@ -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)