Update test_gluon.py

apache · Sep 26, 2019 · 5be5840 · 5be5840
1 parent 04765a3
commit 5be5840
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/python/unittest/test_gluon.py b/tests/python/unittest/test_gluon.py
@@ -473,7 +473,7 @@ def hybrid_forward(self, F, a, b):
         foo1 = FooNested(prefix='foo_nested_hybridized')
         foo1.hybridize()
         foo2 = FooNested(prefix='foo_nested_nohybrid')
-        for _ in range(2):
+        for _ in range(2): # Loop for 2 times to trigger forwarding of the cached version
             out1 = foo1(*arg_inputs)
             out2 = foo2(*arg_inputs)
             if isinstance(out1, tuple):