diff --git a/tests/python/unittest/test_gluon.py b/tests/python/unittest/test_gluon.py index 767b31c23437..bfbe10779b1a 100644 --- 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))