try to fix possible memory leak

apache · May 25, 2021 · 978f97e · 978f97e
1 parent 0b1955b
commit 978f97e
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/tests/python/unittest/test_gluon.py b/tests/python/unittest/test_gluon.py
@@ -2300,7 +2300,6 @@ def forward(self, x):
             check_layer_forward_withinput(net, x)
 
 @pytest.mark.serial
-@pytest.mark.skip("GPU out of memory")
 def test_slice_pooling2d_slice_pooling2d():
     max_pooling = nn.MaxPool2D(strides=(2, 3), padding=(1, 1))
     avg_pooling = nn.AvgPool2D(strides=(2, 2), padding=(1, 1))