大家好,我遇到了一个奇怪的问题。
我一直在使用的两个模型之一有问题。模型将图像数据作为输入并输出操纵杆和键盘信息。
- 一个简单的 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 参数更改为较小的批量大小,但没有任何结果。