Я запускаю чужую программу для глубокого обучения с использованием тензорного потока, и, поскольку я все еще новичок в этом, возникла ошибка исчерпания ресурсов при попытке обучить данные. Я делал то же самое в течение последних месяцев, но никогда не сталкивался с этой проблемой.
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, и та же проблема продолжает происходить. Как мне преодолеть эту проблему?