search

maudzung / SFA3D

Super Fast and Accurate 3D Object Detection based on 3D LiDAR Point Clouds (The PyTorch implementation)
https://github.com/maudzung/Super-Fast-Accurate-3D-Object-Detection
MIT License
1k stars 271 forks source link

Cuda out of memory #49

Closed mouawiya closed 2 years ago

mouawiya commented 2 years ago

hello, i am currently always facing a cuda out of memory error when i run the train.py file, i looked though the internet it said that i needed to lower the batch size but i don't know how to dot that, there isn't like a clear batch size number or value i can decrease, so can you please help me.

urbansound8K commented 2 years ago

python train.py --gpu_idx 0 --batch_size 1

maudzung commented 2 years ago

Thanks @urbansound8K