ztsrxh / RoadBEV

Codes for RoadBEV: road surface reconstruction in Bird's Eye View
MIT License
131 stars 12 forks source link

Is distributed training supported, and how much GPU memory is required for stereo training? #11

Open qiurui4shu opened 5 days ago

ztsrxh commented 4 days ago

The provided code is for single GPU. You can modify it for multi-GPU training. For RoadBEV-stereo, about 22GB if batch_size=4