search

dongwu92 / AutoPortraitMatting

Tensorflow implementation of Automatic Portrait Matting on paper "Automatic Portrait Segmentation for Image Stylization"
Apache License 2.0
479 stars 151 forks source link

ResourceExhaustedError (see above for traceback): OOM when allocating tensor with shape[20,800,600,64] #34

Open wasedawesley opened 7 years ago

wasedawesley commented 7 years ago

Limit: 3406036992 InUse: 2376772608 MaxInUse: 2613990400 NumAllocs: 197 MaxAllocSize: 411041792

W tensorflow/core/common_runtime/bfcallocator.cc:274] ****____***____ W tensorflow/core/common_runtime/bfc_allocator.cc:275] Ran out of memory trying to allocate 2.29GiB. See logs for memory state. W tensorflow/core/framework/op_kernel.cc:993] Resource exhausted: OOM when allocating tensor with shape[20,800,600,64]

I have 4G ram, but nevertheless how much I change the batch size to , it won't work, anyone can help me?

我batch size不管怎么调，始终显示上面的错误，而且batch size越小，INuse的量反而越大

qqsh0214 commented 6 years ago

@wasedawesley Which one do you use? FCN.py or FCN_plus.py?