Spaces:
Running
Running
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)))
|