Update test_gluon.py

apache · Sep 26, 2019 · 04765a3 · 04765a3
1 parent c577029
commit 04765a3
Showing 1 changed file with 8 additions and 7 deletions.
diff --git a/tests/python/unittest/test_gluon.py b/tests/python/unittest/test_gluon.py
@@ -473,13 +473,14 @@ def hybrid_forward(self, F, a, b):
         foo1 = FooNested(prefix='foo_nested_hybridized')
         foo1.hybridize()
         foo2 = FooNested(prefix='foo_nested_nohybrid')
-        out1 = foo1(*arg_inputs)
-        out2 = foo2(*arg_inputs)
-        if isinstance(out1, tuple):
-            for lhs, rhs in zip(out1, out2):
-                assert_almost_equal(lhs.asnumpy(), rhs.asnumpy())
-        else:
-            assert_almost_equal(out1.asnumpy(), out2.asnumpy())
+        for _ in range(2):
+            out1 = foo1(*arg_inputs)
+            out2 = foo2(*arg_inputs)
+            if isinstance(out1, tuple):
+                for lhs, rhs in zip(out1, out2):
+                    assert_almost_equal(lhs.asnumpy(), rhs.asnumpy())
+            else:
+                assert_almost_equal(out1.asnumpy(), out2.asnumpy())
     for do_hybridize in [True, False]:
         foo = FooNested()
         if do_hybridize: