vrvundyala commited on
Commit
46ba4ca
·
1 Parent(s): 9097fb5

siamese network first push

Browse files
app/Hackathon_setup/face_recognition.py CHANGED
@@ -77,7 +77,7 @@ def get_similarity(img1, img2):
77
  # YOUR CODE HERE, load the model
78
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
79
  myModel = Siamese().to(device)
80
- ckpt = torch.load('./siamese_model.t7')
81
  myModel.load_state_dict(ckpt['net_dict'])
82
  myModel.eval()
83
 
 
77
  # YOUR CODE HERE, load the model
78
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
79
  myModel = Siamese().to(device)
80
+ ckpt = torch.load('siamese_model.t7')
81
  myModel.load_state_dict(ckpt['net_dict'])
82
  myModel.eval()
83