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.