ldkong1205 / LaserMix

[CVPR 2023 Highlight] LaserMix for Semi-Supervised LiDAR Semantic Segmentation
https://ldkong.com/LaserMix
Apache License 2.0
272 stars 17 forks source link

Question about the requirement of GPUs #12

Closed Hugo-cell111 closed 8 months ago

Hugo-cell111 commented 8 months ago

Hi! How many GPUs and what kind of GPUs are needed in semi-supervised point-cloud segmentation task?

ldkong1205 commented 8 months ago

Hi @Hugo-cell111, thanks for your interest in this work!

For your question:

Method Backend Memory Training Time FPS
MinkUNet18-W16 torchsparse 3.4 GB - -
MinkUNet34-W32 minkowski engine 11.5 GB 6.5 hours 12.2
MinkUNet34-W32 spconv 6.7 GB 2.0 hours 14.6
Cylinder3D spconv 10.2 GB - -
ldkong1205 commented 8 months ago

Hi @Hugo-cell111, hope the above answer resolves your question. Free feel to let us know if you have further questions!