Тренировочные сбои Tensorflow превышают 10% системной памяти, хотя тренировка с размером пакета 1 - PullRequest
0 голосов
/ 18 июня 2019

Обучение с размером пакета 128/64/32, используемое для простого опустошения памяти графического процессора после нескольких эпох.Однако выполнение стохастического пакетного обучения фактически приводит к тому, что программа застревает на 0% от 1-й эпохи.

    --------------- Epoch 1 ---------------
  0%|                                                                                          | 0/486 [00:00<?, ?it/s]2019-06-18 18:04:58.581233: W T:\src\github\tensorflow\tensorflow\core\framework\allocator.cc:108] Allocation of 1207959552 exceeds 10% of system memory.
2019-06-18 18:04:59.208729: W T:\src\github\tensorflow\tensorflow\core\framework\allocator.cc:108] Allocation of 1207959552 exceeds 10% of system memory.
2019-06-18 18:04:59.827425: W T:\src\github\tensorflow\tensorflow\core\framework\allocator.cc:108] Allocation of 1207959552 exceeds 10% of system memory.
2019-06-18 18:05:00.497830: W T:\src\github\tensorflow\tensorflow\core\framework\allocator.cc:108] Allocation of 1207959552 exceeds 10% of system memory.
2019-06-18 18:05:01.173273: W T:\src\github\tensorflow\tensorflow\core\framework\allocator.cc:108] Allocation of 1207959552 exceeds 10% of system memory.

Используется графический процессор GTX 1080, у кого-нибудь есть идеи?Заранее спасибо.

...