Prediction error

[WhoAmI-kai]

I randomly selected 200 data from Argoverse Motion Forecasting's val for prediction testing, and the predictions were all nan.

The test command is: python test.py -m lanegcn --weight=36.000.ckpt --split=val
Pre-trained models from: http://yun.sfo2.digitaloceanspaces.com/public/lanegcn/36.000.ckpt

The data of the input model is normal:

But the predictions were all nan:

Hope to get your help. @chenyuntc