search

ZikangZhou / HiVT

[CVPR 2022] HiVT: Hierarchical Vector Transformer for Multi-Agent Motion Prediction
https://openaccess.thecvf.com/content/CVPR2022/papers/Zhou_HiVT_Hierarchical_Vector_Transformer_for_Multi-Agent_Motion_Prediction_CVPR_2022_paper.pdf
Apache License 2.0
577 stars 115 forks source link

support pytorch 2.0, lightning 2.0 and pyg 2.3 #47

Open Dinngger opened 9 months ago

Dinngger commented 9 months ago

Tested on Ubuntu 20.04 with CUDA 11.8. Only test eval.py with given checkpoint.