tsaddev commited on
Commit
1e160ee
·
1 Parent(s): 6f858d5

Update app/Hackathon_setup/exp_recognition_model.py

Browse files
app/Hackathon_setup/exp_recognition_model.py CHANGED
@@ -19,9 +19,8 @@ classes = {0: 'ANGER', 1: 'DISGUST', 2: 'FEAR', 3: 'HAPPINESS', 4: 'NEUTRAL', 5:
19
 
20
  # Example Network
21
  class facExpRec(torch.nn.Module):
22
- def __init__(self, out_features=7):
23
  super().__init__()
24
-
25
  self.conv1 = self.convlayer(in_channels=1, out_channels=64, kernel_size=3, max_pool=2)
26
  self.conv2 = self.convlayer(in_channels=64, out_channels=128, kernel_size=3, max_pool=2)
27
  self.conv3 = self.convlayer(in_channels=128, out_channels=512, kernel_size=3, max_pool=2)
 
19
 
20
  # Example Network
21
  class facExpRec(torch.nn.Module):
22
+ def __init__(self, out_features=7):
23
  super().__init__()
 
24
  self.conv1 = self.convlayer(in_channels=1, out_channels=64, kernel_size=3, max_pool=2)
25
  self.conv2 = self.convlayer(in_channels=64, out_channels=128, kernel_size=3, max_pool=2)
26
  self.conv3 = self.convlayer(in_channels=128, out_channels=512, kernel_size=3, max_pool=2)