Tensor OOM when allocating tensor with shape[1,512,480,480]

SUYEgit / Surgery-Robot-Detection-Segmentation

Object detection and segmentation for a surgery robot using Mask-RCNN on Python 3, Keras, and TensorFlow..

Other

181 stars 84 forks source link

Tensor OOM when allocating tensor with shape[1,512,480,480] #22

Open luckystar2794 opened 3 years ago

luckystar2794 commented 3 years ago

May I check with you regarding this configuration why would I face OOM issue?

I'm using GTX 1050. Thanks any advice to resolve this would help me greatly!! I'm new to training model on GPU but I've done the necessary setup for tensorflow to work on gpu. fyi.

luckystar2794 commented 3 years ago

Forgot to mention the resolution of image I'm training with is 1920x1024