如何加速PDMS的计算

你好，非常感谢你们的工作！

在训练的时候遇到一个问题，get_pdm_score_para()函数特别耗时间，导致GPU利用率很低，CPU利用率很高。有没有加速的方法？

谢谢！
