diff options
author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-06-05 21:18:56 +0200 |
---|---|---|
committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-06-05 21:18:56 +0200 |
commit | 425af1bce8362efd97682a5042e76a60bfc28060 (patch) | |
tree | a06bba00f825c0f7cb69a8df2f4d424af8070ef7 /text_recognizer/networks/conformer/depth_wise_conv.py | |
parent | 6e1ad65edd7cbb0f8eb7a48991e9f000f554761d (diff) |
Remove depth wise conv class
Diffstat (limited to 'text_recognizer/networks/conformer/depth_wise_conv.py')
-rw-r--r-- | text_recognizer/networks/conformer/depth_wise_conv.py | 17 |
1 files changed, 0 insertions, 17 deletions
diff --git a/text_recognizer/networks/conformer/depth_wise_conv.py b/text_recognizer/networks/conformer/depth_wise_conv.py deleted file mode 100644 index 1dbd0b8..0000000 --- a/text_recognizer/networks/conformer/depth_wise_conv.py +++ /dev/null @@ -1,17 +0,0 @@ -"""Depthwise 1D convolution.""" -from torch import nn, Tensor - - -class DepthwiseConv1D(nn.Module): - def __init__(self, in_channels: int, out_channels: int, kernel_size: int) -> None: - super().__init__() - self.conv = nn.Conv1d( - in_channels=in_channels, - out_channels=out_channels, - kernel_size=kernel_size, - groups=in_channels, - padding="same", - ) - - def forward(self, x: Tensor) -> Tensor: - return self.conv(x) |