From 4ad6d0af2f4e6868576582e4b86b096a2e95e6fa Mon Sep 17 00:00:00 2001 From: Jordan Gong Date: Thu, 24 Dec 2020 15:42:42 +0800 Subject: Change the usage of layers and reorganize relations of layers 1. Add batch normalization and activation to layers 2. VGGConv2d and FocalConv2d inherits to BasicConv2d; DCGANConvTranspose2d inherits to BasicConvTranspose2d --- models/fpfe.py | 14 ++++++++------ 1 file changed, 8 insertions(+), 6 deletions(-) (limited to 'models/fpfe.py') 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 -- cgit v1.2.3