@@ -1371,8 +1371,8 @@ def test_split_normalization_params(self, norm_layer):
1371
1371
class TestDropBlock :
1372
1372
@pytest .mark .parametrize ("seed" , range (10 ))
1373
1373
@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 ])
1376
1376
@pytest .mark .parametrize ("inplace" , [True , False ])
1377
1377
def test_drop_block (self , seed , dim , p , block_size , inplace ):
1378
1378
torch .manual_seed (seed )
@@ -1439,7 +1439,7 @@ def make_obj(self, dim, p, block_size, inplace, wrap=False):
1439
1439
1440
1440
@pytest .mark .parametrize ("dim" , (2 , 3 ))
1441
1441
@pytest .mark .parametrize ("p" , [0 , 1 ])
1442
- @pytest .mark .parametrize ("block_size" , [5 , 7 , 9 , 11 ])
1442
+ @pytest .mark .parametrize ("block_size" , [5 , 7 ])
1443
1443
@pytest .mark .parametrize ("inplace" , [True , False ])
1444
1444
def test_is_leaf_node (self , dim , p , block_size , inplace ):
1445
1445
op_obj = self .make_obj (dim , p , block_size , inplace , wrap = True )
0 commit comments