about axis-angle output

[amituofo1996]

Hi, it is really a nice work.
I want to get anxis-angle format result.Following is my code.
with torch.no_grad():
pred_rotmat, pred_betas, pred_camera = model(norm_img.to(device))
pred_output = smpl(betas=pred_betas, body_pose=pred_rotmat[:,1:], global_orient=pred_rotmat[:,0].unsqueeze(1), pose2rot=False,return_full_pose=True)
pred_vertices = pred_output.vertices
pred_fullpose = pred_output.full_pose
homogen_matrot = F.pad(pred_rotmat.view(-1,3,3),[0,1])
pred_pose = tgm.rotation_matrix_to_angle_axis(homogen_matrot).contiguous().view(-1, 72)

i got pred_pose as my result,but it is wrong,I don't know why,will somebody could help me!