diff options
Diffstat (limited to 'models/fpfe.py')
| -rw-r--r-- | models/fpfe.py | 14 |
1 files changed, 8 insertions, 6 deletions
diff --git a/models/fpfe.py b/models/fpfe.py index 7d04655..29f7a5f 100644 --- a/models/fpfe.py +++ b/models/fpfe.py @@ -25,13 +25,15 @@ class FrameLevelPartFeatureExtractor(nn.Module): self.max_pool = nn.MaxPool2d(kernel_size=2, stride=2) def forward(self, x): - x = F.leaky_relu(self.focal_conv1(x), inplace=True) - x = F.leaky_relu(self.focal_conv2(x), inplace=True) + x = self.focal_conv1(x) + x = self.focal_conv2(x) x = self.max_pool(x) - x = F.leaky_relu(self.focal_conv3(x), inplace=True) - x = F.leaky_relu(self.focal_conv4(x), inplace=True) + + x = self.focal_conv3(x) + x = self.focal_conv4(x) x = self.max_pool(x) - x = F.leaky_relu(self.focal_conv5(x), inplace=True) - x = F.leaky_relu(self.focal_conv6(x), inplace=True) + + x = self.focal_conv5(x) + x = self.focal_conv6(x) return x |