fix cpu stuff

metrics/metrics.py

@@ -35,7 +35,7 @@ class IdScore(nn.Module):
 		super(IdScore, self).__init__()
 		# print('Loading ResNet ArcFace')
 		self.facenet = Backbone(input_size=112, num_layers=50, drop_ratio=0.6).to(device)
-		self.facenet.load_state_dict(torch.load('./pretrained_models/model_ir_se50.pth',  map_location=torch.device('cpu'))).to(device)
+		self.facenet.load_state_dict(torch.load('./pretrained_models/model_ir_se50.pth',  map_location=torch.device(device)))
 		self.face_pool = torch.nn.AdaptiveAvgPool2d((112, 112))
 		self.facenet.eval()
 		self.cosine_sim = nn.CosineSimilarity(dim=1)