fix example

PaddlePaddle · Mar 16, 2023 · 42e19ad · 42e19ad
1 parent d62762f
commit 42e19ad
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_resnet_v2.py b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_resnet_v2.py
@@ -427,6 +427,7 @@ def test_resnet(self):
 
     def test_resnet_composite(self):
         core._set_prim_backward_enabled(True)
+        core._add_skip_comp_ops("batch_norm")
         static_loss = self.train(to_static=True)
         core._set_prim_backward_enabled(False)
         dygraph_loss = self.train(to_static=False)