OOM
cs19469 opened this issue · comments
hello, i have a question. I want to predict all the pictures of koniq using the trained model. So, I used a loop to process all the pictures in the folder, but there are some problem like this, can you help me?
tensorflow.python.framework.errors_impl.ResourceExhaustedError: OOM when allocating tensor with shape[64,386,514] and type float on /job:localhost/replica:0/task:0/device:GPU:0 by allocator GPU_0_bfc
[[node model_10/pool1_pad/Pad (defined at E:/Graudate/Code/triq-master-play/src/examples/image_quality_prediction.py:21) ]]
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.
[Op:__inference_predict_function_106843]
Hi, you can try to reduce the batch size, e.g., using 4 or 2 or 1. Your GPU memory might not be large enough.
But I didn't do the training, just preidict using the code of example
But I didn't do the training, just preidict using the code of example
I am sorry but I cannot reproduce the error. It ran normally on my PC.
Thank you, I have solved it.