Spaces:
Sleeping
Sleeping
Pallavi Bhoj
commited on
Commit
·
434cd9b
1
Parent(s):
4f5950a
Update face_recognition_model.py
Browse files
app/Hackathon_setup/face_recognition_model.py
CHANGED
@@ -14,7 +14,7 @@ trnscm = transforms.Compose([transforms.Resize((100,100)), transforms.ToTensor()
|
|
14 |
class Siamese(torch.nn.Module):
|
15 |
def __init__(self):
|
16 |
super(Siamese, self).__init__()
|
17 |
-
super(SiameseNetwork, self).__init__()
|
18 |
self.cnn1 = nn.Sequential(
|
19 |
nn.ReflectionPad2d(1),
|
20 |
# Pads the input tensor using the reflection of the input boundary, it similar to the padding.
|
|
|
14 |
class Siamese(torch.nn.Module):
|
15 |
def __init__(self):
|
16 |
super(Siamese, self).__init__()
|
17 |
+
# super(SiameseNetwork, self).__init__()
|
18 |
self.cnn1 = nn.Sequential(
|
19 |
nn.ReflectionPad2d(1),
|
20 |
# Pads the input tensor using the reflection of the input boundary, it similar to the padding.
|