comment debug print

This commit is contained in:
xiaoke@ivgsz_171 2023-02-07 16:02:02 +08:00
parent 31f3b2022e
commit c15030f76a

View File

@ -33,7 +33,7 @@ class SkipPoses(BeforeAfterBase):
def before(self, body_params): def before(self, body_params):
poses = body_params['poses'] poses = body_params['poses']
poses_copy = torch.zeros_like(poses) poses_copy = torch.zeros_like(poses)
print(poses.shape) # print(poses.shape)
poses_copy[..., self.copy_index] = poses[..., self.copy_index] poses_copy[..., self.copy_index] = poses[..., self.copy_index]
body_params['poses'] = poses_copy body_params['poses'] = poses_copy
return body_params return body_params