Исчезает память в Керасе с меньшим CNN - PullRequest
0 голосов
/ 08 мая 2018

Привет всем, у меня странная проблема.

Я получил данные, которые являются изображением, и вывод, который является информацией о джойстике и клавиатурой. Модель, с которой у меня не возникает проблем с нехваткой памяти (и сбоем), это inception_v3, которая намного сложнее, чем простая CNN (по этой причине я думаю, что это странно).

https://github.com/Sentdex/pygta5/blob/efdd66841ff909a027a42ebb48facbd63670e281/models.py#L219

Я пытаюсь применить модель keras, основанную на модели MNIST, и меняю размер входного файла.

https://github.com/keras-team/keras/blob/9c689775a22f9e8647207eeae012978048d44bed/examples/mnist_cnn.py#L47

Вывод перед сбоем это более одного раза. Это начинает эпоху, так что я думаю, что это не проблема с формой данных.

> 2018-05-07 23:03:22.080446: W T:\src\github\tensorflow\tensorflow\core\framework\allocator.cc:101] Allocation of 1037238272 exceeds 10% of system memory.
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...