From 978f97e8cd1d486a05aaa0682b209150e1178dfd Mon Sep 17 00:00:00 2001 From: barry-jin Date: Tue, 25 May 2021 14:48:37 -0700 Subject: [PATCH] try to fix possible memory leak --- tests/python/unittest/test_gluon.py | 1 - 1 file changed, 1 deletion(-) 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))