How much GPU resources are required?

tusen-ai / SST

Code for a series of work in LiDAR perception, including SST (CVPR 22), FSD (NeurIPS 22), FSD++ (TPAMI 23), FSDv2, and CTRL (ICCV 23, oral).

Apache License 2.0

801 stars 102 forks source link

How much GPU resources are required? #58

Closed hailong23-jin closed 2 years ago

hailong23-jin commented 2 years ago

How long does it take to train on waymo dataset? and how much GPU resources are required? Four 3090 or V100? Can NVIDIA TITAN TRX 24G train this model ?

Abyssaledge commented 2 years ago

Thanks for using. The single-frame Waymo model can be trained on 2080Ti with batchsize=1. For the full training dataset and 2x schedule (24 epochs), it takes around 3 days in 8 2080Ti GPUs.