较小的CNN的内存问题

数据挖掘 喀拉斯 麻木的 卷积神经网络 学习
2021-09-18 22:01:08

大家好,我遇到了一个奇怪的问题。

我一直在使用的两个模型之一有问题。模型将图像数据作为输入并输出操纵杆和键盘信息。

  • 一个简单的 CNN,使用 Keras 实现以与 MNIST 数据一起使用,我在其中更改了输入大小;
  • inception_v3(比第一个复杂得多)

在训练第一个模型时,它会引发以下错误:

> 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.

我尝试将 batch_size 参数更改为较小的批量大小,但没有任何结果。

0个回答
没有发现任何回复~