Skip to content

Commit d73b81e

Browse files
committed
Time reduction factor calculation is fixed.
1 parent d4e1fea commit d73b81e

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tensorflow_asr/models/layers/subsampling.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -100,7 +100,7 @@ def __init__(
100100
bias_regularizer=bias_regularizer,
101101
)
102102
self.maxpool2 = tf.keras.layers.MaxPool2D(pool_size=strides, padding="same", name=f"{name}_maxpool_2")
103-
self.time_reduction_factor = self.maxpool1.pool_size[0] + self.maxpool2.pool_size[0]
103+
self.time_reduction_factor = self.maxpool1.pool_size[0] * self.maxpool2.pool_size[0]
104104

105105
def call(
106106
self,
@@ -165,7 +165,7 @@ def __init__(
165165
kernel_regularizer=kernel_regularizer,
166166
bias_regularizer=bias_regularizer,
167167
)
168-
self.time_reduction_factor = self.conv1.strides[0] + self.conv2.strides[0]
168+
self.time_reduction_factor = self.conv1.strides[0] * self.conv2.strides[0]
169169

170170
def call(
171171
self,

0 commit comments

Comments
 (0)