From dedf51f4fcbfb954558e157a0e449fc2ea48e7e1 Mon Sep 17 00:00:00 2001 From: Jordan Gong Date: Wed, 23 Dec 2020 21:06:56 +0800 Subject: Make activation inplace --- models/fpfe.py | 14 +++++++------- 1 file 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 -- cgit v1.2.3