fix depth-wise conv

LLYX · Oct 26, 2020 · 9453375 · 9453375
1 parent 1be7766
commit 9453375
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/conformer/conformer.py b/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
 

diff --git a/setup.py b/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',