fix stride for 1d maxpooling (#5710)

* fix stride for 1d maxpooling

* fix test
This commit is contained in:
Svetlana Dolinina authored and GitHub committed 2021-05-21 11:57:40 +03:00
1 parent af915859e3
commit 8e956654bf
2 files changed
+2 -2

No files matched your search

@@ -36,7 +36,7 @@ class MaxPoolingComponentFrontExtractor(FrontExtractorOp):
mapping_rule = {
'window': np.array([1, 1, 1, kernel], dtype=np.int64),
'stride': np.array([1, 1, stride, stride], dtype=np.int64),
'stride': np.array([1, 1, 1, stride], dtype=np.int64),
'pool_stride': pool_stride,
'pool_step': pool_step,
'pad': np.array([[0, 0], [0, 0], [0, 0], [0, 0]], dtype=np.int64),
@@ -27,7 +27,7 @@ class MaxPoolingComponentFrontExtractorTest(KaldiFrontExtractorTest):
def test_attrs(self):
val_attrs = {
'window': [1, 1, 1, 2],
'stride': [1, 1, 2, 2],
'stride': [1, 1, 1, 2],
'pool_stride': 4,
'pad': [[[0, 0], [0, 0], [0, 0], [0, 0]]]
}