question about vq-vae reconstruction loss

[pangwong]

Hi, I found the velocity calculation in the forward_vel function in losses.py may not use the correct features

def forward_vel(self, motion_pred, motion_gt) : 
    loss = self.Loss(motion_pred[..., 4 : (self.nb_joints - 1) * 3 + 4], motion_gt[..., 4 : (self.nb_joints - 1) * 3 + 4])
    return loss

in the code the data slice is [4:(self.nb_joints - 1) * 3 + 4], which should be rotation independent positions, not the velocity
did I miss something ?