@@ -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