A-bone1 / Attention-ocr-Chinese-Version

Attention OCR Based On Tensorflow
431 stars 141 forks source link

代码跑不起来 #66

Closed FortuneStar closed 4 years ago

FortuneStar commented 4 years ago

(0) Resource exhausted: OOM when allocating tensor with shape[150,600,3] and type float on /job:localhost/replica:0/task:0/device:GPU:0 by allocator GPU_0_bfc [[{{node PreprocessImage/convert_image/Cast}}]] Hint: If you want to see a list of allocated tensors when OOM happens, add report_tensor_allocations_upon_oom to RunOptions for current allocation info.

 [[PreprocessImage/AugmentImage/resize_2/ExpandDims/_1725]]

Hint: If you want to see a list of allocated tensors when OOM happens, add report_tensor_allocations_upon_oom to RunOptions for current allocation info.

(1) Resource exhausted: OOM when allocating tensor with shape[150,600,3] and type float on /job:localhost/replica:0/task:0/device:GPU:0 by allocator GPU_0_bfc [[{{node PreprocessImage/convert_image/Cast}}]] Hint: If you want to see a list of allocated tensors when OOM happens, add report_tensor_allocations_upon_oom to RunOptions for current allocation info.

batchsize已经改成了1还是不行,我显卡是2080 Ti的