| 158 | return out |
| 159 | |
| 160 | class visualConv1D(nn.Module): |
| 161 | def __init__(self): |
| 162 | super(visualConv1D, self).__init__() |
| 163 | self.net = nn.Sequential( |
| 164 | nn.Conv1d(512, 256, 5, stride=1, padding=2), |
| 165 | nn.BatchNorm1d(256), |
| 166 | nn.ReLU(), |
| 167 | nn.Conv1d(256, 128, 1), |
| 168 | ) |
| 169 | |
| 170 | def forward(self, x): |
| 171 | out = self.net(x) |
| 172 | return out |