Henry1iu / TNT-Trajectory-Prediction

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

训练的时候出现Segmentation fault (core dumped) #33

Closed YangQY320 closed 2 years ago

YangQY320 commented 2 years ago

您好!我运行train_tnt.py的时候会在训练一定时间后出现segmentation fault(core dumped)的错误:

[Info: Device_0: train_Ep_11: loss: 4.38811e+02; avg_loss: 4.53619e+02]: 20%|| 11/56 [00:02<00:08, 5.02it/s]Segmentation fault (core dumped)

然后程序就会直接结束,请问您之前有遇到过这样的问题吗?我运行环境下相关库的版本如下: torch 1.8.1+cu111 torch-cluster 1.6.0 torch-geometric 2.1.0.post1 torch-scatter 2.0.9 torch-sparse 0.6.12 torch-spline-conv 1.2.1 torchaudio 0.8.1 torchvision 0.9.1+cu111

YangQY320 commented 2 years ago

但训练vectornet的时候就不会出这样的问题