Make activation inplace

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