使用较小的CNN在卡拉斯耗尽内存

时间:2018-05-07 21:08:39

标签: tensorflow deep-learning keras artificial-intelligence tflearn

大家好,我有一个奇怪的问题。

我得到的数据是图像和输出,它是操纵杆信息和键盘。 我没有内存(和崩溃)问题的模型是inception_v3,它比简单的CNN复杂得多(因为我觉得这很奇怪)。

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

我正在尝试使用基于MNIST的keras模型并更改输入大小。

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.

0 个答案:

没有答案