Skip to content

Commit ebea539

Browse files
authored
speed the dropblock test
1 parent b159f4d commit ebea539

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

test/test_ops.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1371,8 +1371,8 @@ def test_split_normalization_params(self, norm_layer):
13711371
class TestDropBlock:
13721372
@pytest.mark.parametrize("seed", range(10))
13731373
@pytest.mark.parametrize("dim", [2, 3])
1374-
@pytest.mark.parametrize("p", [0, 0.2, 0.5, 0.8])
1375-
@pytest.mark.parametrize("block_size", [5, 7, 9, 11])
1374+
@pytest.mark.parametrize("p", [0, 0.5])
1375+
@pytest.mark.parametrize("block_size", [5, 11])
13761376
@pytest.mark.parametrize("inplace", [True, False])
13771377
def test_drop_block(self, seed, dim, p, block_size, inplace):
13781378
torch.manual_seed(seed)
@@ -1439,7 +1439,7 @@ def make_obj(self, dim, p, block_size, inplace, wrap=False):
14391439

14401440
@pytest.mark.parametrize("dim", (2, 3))
14411441
@pytest.mark.parametrize("p", [0, 1])
1442-
@pytest.mark.parametrize("block_size", [5, 7, 9, 11])
1442+
@pytest.mark.parametrize("block_size", [5, 7])
14431443
@pytest.mark.parametrize("inplace", [True, False])
14441444
def test_is_leaf_node(self, dim, p, block_size, inplace):
14451445
op_obj = self.make_obj(dim, p, block_size, inplace, wrap=True)

0 commit comments

Comments
 (0)