siamese network first push

app/Hackathon_setup/face_recognition.py

@@ -75,7 +75,8 @@ def get_similarity(img1, img2):
     ##########################################################################################
     
     # YOUR CODE HERE, load the model
-    myModel = Siamese().cuda()
+    device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+    myModel = Siamese().to(device)
     ckpt = torch.load('./siamese_model.t7')
     myModel.load_state_dict(ckpt['net_dict'])
     myModel.eval()