Как я могу исправить мой ресурс исчерпал ошибку Tensorflow (ЦП) - PullRequest
0 голосов
/ 09 января 2020

Я запускаю чужую программу для глубокого обучения с использованием тензорного потока, и, поскольку я все еще новичок в этом, возникла ошибка исчерпания ресурсов при попытке обучить данные. Я делал то же самое в течение последних месяцев, но никогда не сталкивался с этой проблемой.

ResourceExhaustedError: OOM when allocating tensor with shape[30000,512] and type float on /job:localhost/replica:0/task:0/device:CPU:0 by allocator cpu     [[{{node training/RMSprop/mul_2}} = Mul[T=DT_FLOAT,
_device="/job:localhost/replica:0/task:0/device:CPU:0"](RMSprop/lr/read, training/RMSprop/gradients/dense_1/MatMul_grad/MatMul_1)]] 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.

Из кода:

num_cores = 28
GPU = False
CPU = True
if GPU:
    num_GPU = 1
    num_CPU = 1
if CPU:
    num_CPU = 28
    num_GPU = 0

Я пытался изменить плотные слои с 1024 до 512, и когда я пытался найти inte rnet для этой проблемы они говорят, чтобы изменить размер партии, и я сделал от 100 до 16, а моя эпоха от 100 до 70, и та же проблема продолжает происходить. Как мне преодолеть эту проблему?

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...