Bug
In sam_3d_body/models/heads/mhr_head.py, the mhr_forward() method has a comment # B x 127 on the line where full_pose_params is assembled:
full_pose_params = torch.cat([global_trans * 10, global_rot, body_pose_params], dim=1) # B x 127
The actual dimension is 136 (3 global_trans + 3 global_rot + 130 body_pose_params), not 127. The MHR paper (arXiv:2511.15586, Section 3.1) confirms n_pose = 136.
127 is the number of joints in the MHR skeleton, not the number of pose parameters.
Suggested fix
full_pose_params = torch.cat([global_trans * 10, global_rot, body_pose_params], dim=1) # B x 136 (3 trans + 3 rot + 130 body_pose)
Found during a documentation audit of the SAM-3D-Body / MHR parameter pipeline.
Bug
In
sam_3d_body/models/heads/mhr_head.py, themhr_forward()method has a comment# B x 127on the line wherefull_pose_paramsis assembled:The actual dimension is 136 (3 global_trans + 3 global_rot + 130 body_pose_params), not 127. The MHR paper (arXiv:2511.15586, Section 3.1) confirms
n_pose = 136.127 is the number of joints in the MHR skeleton, not the number of pose parameters.
Suggested fix
Found during a documentation audit of the SAM-3D-Body / MHR parameter pipeline.