@@ -57,8 +57,6 @@ def __init__(self, channels=None, kernel_size=3, gamma=2, beta=1):
57
57
t = int (abs (math .log (channels , 2 ) + beta ) / gamma )
58
58
kernel_size = max (t if t % 2 else t + 1 , 3 )
59
59
60
- print ('florg' , kernel_size )
61
-
62
60
self .avg_pool = nn .AdaptiveAvgPool2d (1 )
63
61
self .conv = nn .Conv1d (1 , 1 , kernel_size = kernel_size , padding = (kernel_size - 1 ) // 2 , bias = False )
64
62
@@ -86,12 +84,12 @@ class CecaModule(nn.Module):
86
84
(parameter size, throughput,latency, etc)
87
85
88
86
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
90
88
for actual calculations according to channel.
91
89
gamma, beta: when channel is given parameters of mapping function
92
90
refer to original paper https://arxiv.org/pdf/1910.03151.pdf
93
91
(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)
95
93
"""
96
94
97
95
def __init__ (self , channels = None , kernel_size = 3 , gamma = 2 , beta = 1 ):
0 commit comments