Hi,

Please refer to the comments in vis.ipynb, which tell that:

"data.pkl can be collected using the evaluation script, simply by manually saving data after testing.
Important things for visualization: input pos (x_in), target pos (x_end), the prediction (x_gmn)
and the cfg (kinematics decomposition)."

You can load your model, infer the results required above, and save them into data.pkl.

Let me know if you have further questions!