Update app/Hackathon_setup/exp_recognition_model.py

app/Hackathon_setup/exp_recognition_model.py

@@ -20,9 +20,8 @@ class facExpRec(torch.nn.Module):
         self.conv1 = self.convlayer(in_channels=1, out_channels=64, kernel_size=3)
         self.conv2 = self.convlayer(in_channels=64, out_channels=128, kernel_size=5)
         self.conv3 = self.convlayer(in_channels=128, out_channels=512, kernel_size=3)
-        self.conv4 = self.convlayer(in_channels=512, out_channels=512, kernel_size=3, max_pool=1)
 
-        self.fc1 = self.fclayer(512, 256)
+        self.fc1 = self.fclayer(512*3*3, 256)
         self.fc2 = self.fclayer(256, 512)
         self.fc3 = nn.Linear(512, out_features)
     
@@ -48,8 +47,7 @@ class facExpRec(torch.nn.Module):
         x = self.conv1(x)
         x = self.conv2(x)
         x = self.conv3(x)
-        x = self.conv4(x)
-        x = x.view(-1, 512)
+        x = x.view(-1, 512*3*3)
         x = self.fc1(x)
         x = self.fc2(x)
         x = self.fc3(x)