Update i6_models/parts/frontend/generic_frontend.py

Co-authored-by: SimBe195 <[email protected]>
rwth-i6 · Nov 10, 2023 · 261b4f6 · 261b4f6
1 parent ad17746
commit 261b4f6
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/i6_models/parts/frontend/generic_frontend.py b/i6_models/parts/frontend/generic_frontend.py
@@ -176,9 +176,9 @@ def __init__(self, model_cfg: GenericFrontendV1Config):
                 )
                 last_feat_dim = calculate_output_dim(
                     in_dim=last_feat_dim,
-                    filter_size=pool_kernel_size if isinstance(pool_kernel_size, int) else pool_kernel_size[1],
-                    stride=pool_stride if isinstance(pool_stride, int) else pool_stride[1],
-                    padding=pool_padding if isinstance(pool_padding, int) else pool_padding[1],
+                    filter_size=pool_kernel_size[1],
+                    stride=pool_stride[1] or pool_kernel_size[1],
+                    padding=pool_padding[1],
                 )
                 pool_layer_index += 1