search

matterport / Mask_RCNN

Mask R-CNN for object detection and instance segmentation on Keras and TensorFlow
Other
24.55k stars 11.69k forks source link

Train Ballon.py error: OOM when allocating tensor with shape... #964

Open darvida opened 6 years ago

darvida commented 6 years ago

Hey when i run the Ballon.py train command" python balloon.py train --dataset=balloon1 --weights=coco" in command prompt to train on the balloon dataset i get the following error: https://imgur.com/a/GDDv39U I have tried to change some variables in the "Mask_RCNN\mrcnn\config.py" file but i still get the error. Have abyone else encountered this issue?

wenshiqi0 commented 5 years ago

decrease your IMAGES_PER_GPU

athulpa commented 5 years ago

I have tried that. IMAGES_PER_GPU = 1 has been set. Yet, the issue still persists. Note: I have only 2GB of GPU RAM (System RAM=8GB). Is there a way I can reduce the requirements of MRCNN further? Can I downgrade the resolution from 1024? Please help.

yanzistudio commented 5 years ago

I have tried that. IMAGES_PER_GPU = 1 has been set. Yet, the issue still persists. Note: I have only 2GB of GPU RAM (System RAM=8GB). Is there a way I can reduce the requirements of MRCNN further? Can I downgrade the resolution from 1024? Please help.

Hi @athulpa, did you solve your problem? I am also thinking about downgrade the resolution and may need some advice. Thanks!

klrc commented 5 years ago

hi, i tried this and it works: #44