search

Henry1iu / TNT-Trajectory-Prediction

A Unofficial Pytorch Implementation of TNT: Target-driveN Trajectory Prediction
504 stars 95 forks source link

TNT.py中的traj_selection耗时太长 #45

Open rjycty opened 1 year ago

rjycty commented 1 year ago

您好： 我这边在服务器上用较大的数据集进行训练时，发现模型在跑traj_selection这部分时会花费大量时间，使得模型训练特别慢，是因为多重for的原因吗，还是其他原因？ 有没有提高这部分运行速度的好方法，谢谢！

飞书20230322-195405

2023-03-22_19-56