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('
|
| 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 |
|