search

nachiket92 / PGP

Code for "Multimodal Trajectory Prediction Conditioned on Lane-Graph Traversals," CoRL 2021.
https://proceedings.mlr.press/v164/deo22a.html
MIT License
217 stars 36 forks source link

Encounter the RuntimeError when running train.py #16

Closed Ianpengg closed 2 years ago

Ianpengg commented 2 years ago

Hello these days I've tried this code many times But when I run the train.py it always crashed at the beginning I follow the all instructions in your README.md So I'm confused if there is anything I need to setting?

I run this command and use the original .yml file without modified Hope you can help me~ Many thanks python train.py -c configs/pgp_gatx2_lvm_traversal.yml -r /media/ee904/Data_stored/data/Nuscenes -d /media/ee904/Data_stored/data/Nuscenes/preprocess -o /media/ee904/Data_stored/PGP/results -n 100

image

image

Ianpengg commented 2 years ago

I tried to print out the predictions in the trainer.py and I found there is something strange for the 'pi' value is it normal to display nan value? image

nachiket92 commented 2 years ago

@Ianpengg were you able to figure out what the issue was?

Ianpengg commented 2 years ago

Yes I think it's because the value of op_masks I tried to print the output of pi but found nan, so I trace back and find if I print op_masks there would be some -inf value pop up so I tried to add one line of code to convert the inf to a big number What's your opinion for these type of bug? image

this is the result if I print the op_masks image