Wrap Conv1d no bias layer

1 files changed, 7 insertions, 9 deletions

diff --git a/models/tfa.py b/models/tfa.py
index 2e4e656..72bfcfb 100644
--- a/models/tfa.py
+++ b/models/tfa.py
@@ -3,6 +3,8 @@ import copy
 import torch
 from torch import nn as nn
 
+from models.layers import BasicConv1d
+
 
 class TemporalFeatureAggregator(nn.Module):
     def __init__(
@@ -17,11 +19,9 @@ class TemporalFeatureAggregator(nn.Module):
 
         # MTB1
         conv3x1 = nn.Sequential(
-            nn.Conv1d(in_channels, hidden_dim,
-                      kernel_size=3, padding=1, bias=False),
+            BasicConv1d(in_channels, hidden_dim, kernel_size=3, padding=1),
             nn.LeakyReLU(inplace=True),
-            nn.Conv1d(hidden_dim, in_channels,
-                      kernel_size=1, padding=0, bias=False)
+            BasicConv1d(hidden_dim, in_channels, kernel_size=1, padding=0)
         )
         self.conv1d3x1 = self._parted(conv3x1)
         self.avg_pool3x1 = nn.AvgPool1d(kernel_size=3, stride=1, padding=1)
@@ -29,11 +29,9 @@ class TemporalFeatureAggregator(nn.Module):
 
         # MTB2
         conv3x3 = nn.Sequential(
-            nn.Conv1d(in_channels, hidden_dim,
-                      kernel_size=3, padding=1, bias=False),
+            BasicConv1d(in_channels, hidden_dim, kernel_size=3, padding=1),
             nn.LeakyReLU(inplace=True),
-            nn.Conv1d(hidden_dim, in_channels,
-                      kernel_size=3, padding=1, bias=False)
+            BasicConv1d(hidden_dim, in_channels, kernel_size=3, padding=1)
         )
         self.conv1d3x3 = self._parted(conv3x3)
         self.avg_pool3x3 = nn.AvgPool1d(kernel_size=5, stride=1, padding=2)
@@ -76,4 +74,4 @@ class TemporalFeatureAggregator(nn.Module):
 
         # Temporal Pooling
         ret = (feature3x1 + feature3x3).max(-1)[0]
-        return ret
\ No newline at end of file
+        return ret