Оценка моего CNN остановилась с сообщением «OOM при выделении тензора с формой [3,3,256,512]» - PullRequest
Новая
       25

Оценка моего CNN остановилась с сообщением «OOM при выделении тензора с формой [3,3,256,512]»

0 голосов
/ 19 апреля 2019

Я создаю CNN с 5 слоями, 100 эпохами, 32 размерами партии и обучаю его с помощью tenorflow-gpu на устройстве с 16-Гбайтной оперативной памятью, 4-гигабайтной видеокартой NVIDIA GTX с возможностью вычислений 5 cuda.учебный процесс прошел успешно и быстро с 80% тренировочных данных.Моя проблема возникает, когда я хочу оценить оставшиеся 20% невидимых данных изображений в той же сети.Процесс оценки идет очень медленно и останавливается с сообщением: 

    ResourceExhaustedError (see above for traceback): OOM when allocating tensor with shape[3,3,256,512] and type float on /job:localhost/replica:0/task:0/device:GPU:0 by allocator GPU_0_bfc
 [[Node: conv2d_5_122/kernel/Assign = Assign[T=DT_FLOAT, use_locking=true, validate_shape=true, _device="/job:localhost/replica:0/task:0/device:GPU:0"](conv2d_5_122/kernel, conv2d_5_122/random_uniform)]]
    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.

Может кто-нибудь, пожалуйста, помогите мне, я путаюсь.Спасибо.

...