How much memory needed for sem_seg training

charlesq34 / pointnet

PointNet: Deep Learning on Point Sets for 3D Classification and Segmentation

Other

4.64k stars 1.44k forks source link

How much memory needed for sem_seg training #302

Closed AbdouBou closed 1 year ago

AbdouBou commented 1 year ago

I'm trying to train the sem_seg model as indicated in the read-me file.

I'm getting a "Stopped process" indication. At the same time i can see that it occurs exacty when the RAM is full on the monitor.

My question is how much RAM is needed for training without RAM limitation. and where to set a RAM limitation if possible?

rohithsaro commented 1 year ago

Hi @AbdouBou When using the GPU for training it takes up at least 10 GB. However you can try reducing the the batch size according to your specifications. It should be fine after that.

Best,

Rohith

AbdouBou commented 1 year ago

Hi @rohithsaro Ok, thank you for answering.