diff --git a/python/paddle/fluid/contrib/slim/tests/test_graph_wrapper.py b/python/paddle/fluid/contrib/slim/tests/test_graph_wrapper.py index 5340f36196..430ac569f6 100644 --- a/python/paddle/fluid/contrib/slim/tests/test_graph_wrapper.py +++ b/python/paddle/fluid/contrib/slim/tests/test_graph_wrapper.py @@ -37,6 +37,7 @@ def residual_block(num): 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)