Tejaswini2403 commited on
Commit
bb71f7b
·
1 Parent(s): 9dfdffe
app/Hackathon_setup/exp_recognition_model.py CHANGED
@@ -15,7 +15,7 @@ classes = {0: 'ANGER', 1: 'DISGUST', 2: 'FEAR', 3: 'HAPPINESS', 4: 'NEUTRAL', 5:
15
  # Example Network
16
  class facExpRec(torch.nn.Module):
17
  def __init__(self):
18
- super(Model, self).__init__()
19
  self.conv1 = nn.Conv2d(in_channels=1, out_channels=16, kernel_size=3)
20
  self.conv2 = nn.Conv2d(in_channels=16, out_channels=64, kernel_size=3)
21
  self.conv3 = nn.Conv2d(in_channels=64, out_channels=128, kernel_size=3)
 
15
  # Example Network
16
  class facExpRec(torch.nn.Module):
17
  def __init__(self):
18
+ super(facExpRec, self).__init__()
19
  self.conv1 = nn.Conv2d(in_channels=1, out_channels=16, kernel_size=3)
20
  self.conv2 = nn.Conv2d(in_channels=16, out_channels=64, kernel_size=3)
21
  self.conv3 = nn.Conv2d(in_channels=64, out_channels=128, kernel_size=3)