fix depth-wise conv

Author: lucidrains

conformer/conformer.py

@@ -37,12 +37,10 @@ def __init__(self, chan_in, chan_out, kernel_size, padding):
         super().__init__()
         self.padding = padding
         self.conv = nn.Conv1d(chan_in, chan_out, kernel_size, groups = chan_in)
-        self.conv_out = nn.Conv1d(chan_out, chan_out, 1)
 
     def forward(self, x):
         x = F.pad(x, self.padding)
-        x = self.conv(x)
-        return self.conv_out(x)
+        return self.conv(x)
 
 # main class
 

setup.py

@@ -3,7 +3,7 @@
 setup(
   name = 'conformer',
   packages = find_packages(),
-  version = '0.0.3',
+  version = '0.1.0',
   license='MIT',
   description = 'The convolutional module from the Conformer paper',
   author = 'Phil Wang',