Tsinghua-MARS-Lab/DenseTNT

Training time on Waymo

ShoufaChen opened this issue · 4 comments

ShoufaChen commented

Hi,

Thanks for your awesome work.

How long did you train DenseTNT on the Waymo dataset? And how many GPUs (type) did you use?

GentleSmile commented

About 1-2 days with 8 GPUs. For each scene, we randomly select an agent to predict instead of predicting all agents.

ShoufaChen commented

What's the GPU type?

GentleSmile commented

2080Ti

ShoufaChen commented

Thanks.