Does the code support rotations as input instead of joint positions? If so, could you give the instructions for training?