rename args convmixer

Author: ayasyrev

model_constructor/convmixer.py

@@ -42,15 +42,15 @@ def ConvMixerOriginal(dim, depth,
 class ConvLayer(nn.Sequential):
     """Basic conv layers block"""
 
-    def __init__(self, ch_in, ch_out, kernel_size, stride=1,
+    def __init__(self, in_channels, out_channels, kernel_size, stride=1,
                  act_fn=nn.GELU(), padding=0, groups=1,
                  bn_1st=False, pre_act=False):
 
-        conv_layer = [('conv', nn.Conv2d(ch_in, ch_out, kernel_size, stride=stride,
+        conv_layer = [('conv', nn.Conv2d(in_channels, out_channels, kernel_size, stride=stride,
                                          padding=padding, groups=groups))]
         act_bn = [
             ('act_fn', act_fn),
-            ('bn', nn.BatchNorm2d(ch_out))
+            ('bn', nn.BatchNorm2d(out_channels))
         ]
         if bn_1st:
             act_bn.reverse()