From b7db891e0756fb490466246cf802358b1265a0c9 Mon Sep 17 00:00:00 2001
From: Jordan Gong <jordan.gong@protonmail.com>
Date: Wed, 23 Dec 2020 20:15:50 +0800
Subject: Wrap Conv1d no bias layer

---
 models/layers.py | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

(limited to 'models/layers.py')

diff --git a/models/layers.py b/models/layers.py
index a0e35f0..2be93ad 100644
--- a/models/layers.py
+++ b/models/layers.py
@@ -24,3 +24,19 @@ class FocalConv2d(nn.Module):
         z = x.split(split_size, dim=2)
         z = torch.cat([self.conv(_) for _ in z], dim=2)
         return z
+
+
+class BasicConv1d(nn.Module):
+    def __init__(
+            self,
+            in_channels: int,
+            out_channels: int,
+            kernel_size: Union[int, Tuple[int]],
+            **kwargs
+    ):
+        super(BasicConv1d, self).__init__()
+        self.conv = nn.Conv1d(in_channels, out_channels, kernel_size,
+                              bias=False, **kwargs)
+
+    def forward(self, x):
+        return self.conv(x)
-- 
cgit v1.2.3