From f9566c7027eacec7bf1a4720f94cbdf1445457ed Mon Sep 17 00:00:00 2001 From: "shouzhou.bx" Date: Tue, 20 Sep 2022 16:10:23 +0800 Subject: [PATCH] fix face 2d keypoints devices bug --- easycv/models/loss/face_keypoint_loss.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/easycv/models/loss/face_keypoint_loss.py b/easycv/models/loss/face_keypoint_loss.py index 8d4a80c5..52483d0a 100644 --- a/easycv/models/loss/face_keypoint_loss.py +++ b/easycv/models/loss/face_keypoint_loss.py @@ -54,7 +54,7 @@ class WingLossWithPose(nn.Module): self.part_weight = None if part_weight is not None: - self.part_weight = torch.from_numpy(part_weight).cuda() + self.part_weight = torch.from_numpy(part_weight) def forward(self, pred, target, pose): weight = 5.0 * (1.0 - torch.cos(pose * np.pi / 180.0)) + 1.0