OpenDriveLab / ViDAR

[CVPR 2024 Highlight] Visual Point Cloud Forecasting
https://arxiv.org/abs/2312.17655
Apache License 2.0
235 stars 15 forks source link

can a single 3090 be used to pretrain ViDAR? #29

Closed renyuanzhe closed 1 month ago

tomztyang commented 1 month ago

A100 is required to reproduce the results in our paper. Maybe you can try to rescale the image, try smaller BEV resolution, less encoder or decoder layers to reduce the GPU memory cost like Issue 17, but with some performance sacrifice I guess.

Best, Zetong