diff options
| -rw-r--r-- | models/fpfe.py | 14 |
1 files changed, 7 insertions, 7 deletions
diff --git a/models/fpfe.py b/models/fpfe.py index 4a5cde8..7d04655 100644 --- a/models/fpfe.py +++ b/models/fpfe.py @@ -25,13 +25,13 @@ 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)) - x = F.leaky_relu(self.focal_conv2(x)) + x = F.leaky_relu(self.focal_conv1(x), inplace=True) + x = F.leaky_relu(self.focal_conv2(x), inplace=True) x = self.max_pool(x) - x = F.leaky_relu(self.focal_conv3(x)) - x = F.leaky_relu(self.focal_conv4(x)) + x = F.leaky_relu(self.focal_conv3(x), inplace=True) + x = F.leaky_relu(self.focal_conv4(x), inplace=True) x = self.max_pool(x) - x = F.leaky_relu(self.focal_conv5(x)) - x = F.leaky_relu(self.focal_conv6(x)) + x = F.leaky_relu(self.focal_conv5(x), inplace=True) + x = F.leaky_relu(self.focal_conv6(x), inplace=True) - return x
\ No newline at end of file + return x |