fix test_graph_wrapper failure on cudnnv7, test=develop (#23451)

5 years ago · 9bc223c8a2
parent c706ff20a3
commit 9bc223c8a2
1 changed files with 1 additions and 0 deletions
--- 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)