alfredgu001324 / MapUncertaintyPrediction

[CVPR 2024 Award Candidate] Producing and Leveraging Online Map Uncertainty in Trajectory Prediction
https://arxiv.org/abs/2403.16439
Apache License 2.0
154 stars 12 forks source link

Are all experiments done on a single gpu? #9

Closed JT-Sun closed 3 months ago

JT-Sun commented 3 months ago

Thanks for your excellent work, Are all your map training and verification experiments done on a single gpu? Do you have multiple gpu versions?

alfredgu001324 commented 3 months ago

Hi, thanks for your interest in our work! Unfortunately, I do not have a multi-GPU script. All of our experiments are done on a single GPU due to compute resource limitation. But I would encourage looking out the original repo of these works, I believe they do provide scripts for multi-GPU setting!