fix (nn/conv): Fix regression introduced in #1017

Xilinx · Sep 3, 2024 · 3887f3b · 3887f3b
1 parent c096d7b
commit 3887f3b
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/brevitas/nn/quant_conv.py b/src/brevitas/nn/quant_conv.py
@@ -46,7 +46,7 @@ def __init__(
             dtype: Optional[torch.dtype] = None,
             **kwargs) -> None:
         # avoid an init error in the super class by setting padding to 0
-        if padding_mode == 'zeros' and padding == 'same' and any(map(lambda x: x > 1, stride)):
+        if padding_mode == 'zeros' and padding == 'same' and any(map(lambda x: x > 1, list(stride))):
             padding = 0
             is_same_padded_strided = True
         else:
@@ -132,7 +132,7 @@ def __init__(
             dtype: Optional[torch.dtype] = None,
             **kwargs) -> None:
         # avoid an init error in the super class by setting padding to 0
-        if padding_mode == 'zeros' and padding == 'same' and any(map(lambda x: x > 1, stride)):
+        if padding_mode == 'zeros' and padding == 'same' and any(map(lambda x: x > 1, list(stride))):
             padding = 0
             is_same_padded_strided = True
         else:
@@ -220,7 +220,7 @@ def __init__(
             dtype: Optional[torch.dtype] = None,
             **kwargs) -> None:
         # avoid an init error in the super class by setting padding to 0
-        if padding_mode == 'zeros' and padding == 'same' and any(map(lambda x: x > 1, stride)):
+        if padding_mode == 'zeros' and padding == 'same' and any(map(lambda x: x > 1, list(stride))):
             padding = 0
             is_same_padded_strided = True
         else: