Skip to content

Commit d725991

Browse files
committed
Remove debug print from ECA module
1 parent 2a7d256 commit d725991

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

timm/models/layers/eca.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -57,8 +57,6 @@ def __init__(self, channels=None, kernel_size=3, gamma=2, beta=1):
5757
t = int(abs(math.log(channels, 2) + beta) / gamma)
5858
kernel_size = max(t if t % 2 else t + 1, 3)
5959

60-
print('florg', kernel_size)
61-
6260
self.avg_pool = nn.AdaptiveAvgPool2d(1)
6361
self.conv = nn.Conv1d(1, 1, kernel_size=kernel_size, padding=(kernel_size - 1) // 2, bias=False)
6462

@@ -86,12 +84,12 @@ class CecaModule(nn.Module):
8684
(parameter size, throughput,latency, etc)
8785
8886
Args:
89-
channel: Number of channels of the input feature map for use in adaptive kernel sizes
87+
channels: Number of channels of the input feature map for use in adaptive kernel sizes
9088
for actual calculations according to channel.
9189
gamma, beta: when channel is given parameters of mapping function
9290
refer to original paper https://arxiv.org/pdf/1910.03151.pdf
9391
(default=None. if channel size not given, use k_size given for kernel size.)
94-
k_size: Adaptive selection of kernel size (default=3)
92+
kernel_size: Adaptive selection of kernel size (default=3)
9593
"""
9694

9795
def __init__(self, channels=None, kernel_size=3, gamma=2, beta=1):

0 commit comments

Comments
 (0)