search

filaPro / oneformer3d

[CVPR2024] OneFormer3D: One Transformer for Unified Point Cloud Segmentation
Other
351 stars 32 forks source link

Any proposal for lightweight Model #27

Closed Beiliergo closed 8 months ago

Beiliergo commented 9 months ago

Congratulations on achieving such outstanding results. I have tried to train the model, unfortunately, it seems that our training server has not been updated for a long time. Each GPU has only 12GB of memory. Obviously model that cannot be trained in this hardware. At the same time, we hope to run the model on a single card. We can accept a decrease in performance while minimizing memory usage as much as possible. Can you provide a configuration file for a lightweight model? Any suggestions are welcome.

filaPro commented 9 months ago

Are you training on S3DIS ot on ScanNet? You can try to sample less points as an input here. This PointSample_ can also be added to other configs.