remove same case for padding_mode to solve #23 (comment) and #23 (com…

…ment)
deel-ai · Dec 10, 2024 · a0a81da · a0a81da
1 parent a8721bb
commit a0a81da
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 11 deletions.
diff --git a/deel/torchlip/modules/pooling.py b/deel/torchlip/modules/pooling.py
@@ -232,4 +232,3 @@ def forward(self, input: torch.Tensor) -> torch.Tensor:
 
     def vanilla_export(self):
         return self
-
diff --git a/deel/torchlip/utils/lconv_norm.py b/deel/torchlip/utils/lconv_norm.py
@@ -41,11 +41,7 @@ def compute_lconv_coef_1d(
     stride = strides[0]
     k1 = kernel_size[0]
 
-    if (
-        (padding_mode in ["zeros", "same"])
-        and (stride == 1)
-        and (input_shape is not None)
-    ):
+    if (padding_mode in ["zeros"]) and (stride == 1) and (input_shape is not None):
         # See https://arxiv.org/abs/2006.06520
         in_l = input_shape[-1]
         k1_div2 = (k1 - 1) / 2
@@ -67,11 +63,7 @@ def compute_lconv_coef(
     stride = np.prod(strides)
     k1, k2 = kernel_size
 
-    if (
-        (padding_mode in ["zeros", "same"])
-        and (stride == 1)
-        and (input_shape is not None)
-    ):
+    if (padding_mode in ["zeros"]) and (stride == 1) and (input_shape is not None):
         h, w = input_shape[-2:]
         k1_div2 = (k1 - 1) / 2
         k2_div2 = (k2 - 1) / 2
Original file line number	Diff line number	Diff line change
Expand Up		@@ -232,4 +232,3 @@ def forward(self, input: torch.Tensor) -> torch.Tensor:

		def vanilla_export(self):
		return self