Update models/models/detectors/pam.py

models/models/detectors/pam.py

@@ -135,9 +135,9 @@ class PoseAnythingModel(BasePose):
         """Defines the computation performed at every call."""
         str_dict = json.loads(str)
 
-        str_dict["img_q"] = torch.tensor(str_dict["img_q"], dtype=torch.float32)#.cuda()
-        str_dict["target_weight_s"] = torch.tensor(str_dict["target_weight_s"], dtype=torch.float32)#.cuda()
-        str_dict["target_s"] = torch.tensor(str_dict["target_s"], dtype=torch.float32)#.cuda()
+        str_dict["img_q"] = torch.tensor(str_dict["img_q"], dtype=torch.float32).cuda()
+        str_dict["target_weight_s"] = torch.tensor(str_dict["target_weight_s"], dtype=torch.float32).cuda()
+        str_dict["target_s"] = torch.tensor(str_dict["target_s"], dtype=torch.float32).cuda()
         
         str_dict['img_metas'][0]['sample_joints_3d'][0] = torch.tensor(str_dict['img_metas'][0]['sample_joints_3d'][0])
         str_dict['img_metas'][0]['query_joints_3d'] = torch.tensor(str_dict['img_metas'][0]['query_joints_3d'])