fix stride for 1d maxpooling (#5710)
* fix stride for 1d maxpooling * fix test
This commit is contained in:
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]]]
|
||||
}
|
||||
|
||||
Reference in new issue
Block a user