sid-reddy-krishna commited on
Commit
b27a86b
·
1 Parent(s): c1bc8a5
app/Hackathon_setup/face_recognition_model.py CHANGED
@@ -11,7 +11,7 @@ from torchvision import transforms
11
  trnscm = transforms.Compose([transforms.Resize((100,100)), transforms.ToTensor()])
12
 
13
  ##Example Network
14
- class Siamese(torch.nn.Module):
15
  def __init__(self):
16
  super(SiameseNetwork, self).__init__()
17
  self.cnn1 = nn.Sequential(
 
11
  trnscm = transforms.Compose([transforms.Resize((100,100)), transforms.ToTensor()])
12
 
13
  ##Example Network
14
+ class SiameseNetwork(torch.nn.Module):
15
  def __init__(self):
16
  super(SiameseNetwork, self).__init__()
17
  self.cnn1 = nn.Sequential(