Pallavi Bhoj commited on
Commit
d01b65d
·
1 Parent(s): 405f037

Update face_recognition.py

Browse files
app/Hackathon_setup/exp_recognition_model.py CHANGED
@@ -36,7 +36,7 @@ class facExpRec(torch.nn.Module):
36
  self.pool = nn.MaxPool2d(kernel_size=2) # Max pooling layer with filter size 2x2
37
 
38
 
39
- def forward(self, x):
40
  x = self.pool(F.relu(self.conv1(x)))
41
  x = self.pool(F.relu(self.conv2(x)))
42
  x = self.pool(F.relu(self.conv3(x)))
 
36
  self.pool = nn.MaxPool2d(kernel_size=2) # Max pooling layer with filter size 2x2
37
 
38
 
39
+ def forward(self, x):
40
  x = self.pool(F.relu(self.conv1(x)))
41
  x = self.pool(F.relu(self.conv2(x)))
42
  x = self.pool(F.relu(self.conv3(x)))