Encountered some error when evaluating

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

Encountered some error when evaluating #14

Closed Ianpengg closed 2 years ago

Ianpengg commented 2 years ago

Hello thank you for the great work I have some problems when I try to evaluate the metrics. I use the CUDA11.0 and the pytorch 1.7.1 conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=11.0 -c pytorch

And I download the trained weights

I load the weight as -w /media/ee904/Data_stored/PGP/PGP_lr-scheduler/archive/data.pkl then the error pop up I think it's about the weight format.. Can you help me? Or should I need to use the older version of CUDA?

Ianpengg commented 2 years ago

I think there could be different versions between train and test

bkwmartens commented 2 years ago

Loading the .tar file itself as weights should solve your problem.

nachiket92 commented 2 years ago

Yes, please directly pass the .tar file as the weights. -w /media/ee904/Data_stored/PGP/PGP_lr-scheduler.tar