train_PPT problem

[subetter]

Hello,
Many thanks for your excellent work! I am currently attempting to reproduce your results.

When I execute run.sh, I encounter the following error:

After debugging, it seems that the sublists within traj_new have inconsistent lengths, which prevents it from being converted into a NumPy array.

Could you please provide guidance on how to resolve this issue? Thank you very much!

[Xiaotong-Lin]

Hi @subetter ,
Thanks for your interest! You can try adding dtype=object to the np.array() function on lines 82, 83, and 87 in dataset_loader.py. For example, change np.array(traj_new) to np.array(traj_new, dtype=object). This should resolve the issue.