神经网络学习----训练模型过程中的内存不足错误

如果你的计算机GPU很差(比如我的就很差很差),在训练模型时你会发现刚开始训练正常,但后面就会报下图的错误。
神经网络学习----训练模型过程中的内存不足错误_第1张图片
错误:numpy.core._exceptions.MemoryError: Unable to allocate 4.12 MiB for an array with shape (600, 600, 3) and data type float32

该错误是内存不足导致训练不能继续,我的bathsize==1了已经,但还是不行。真不知道这么垃圾的电脑怎么做神经网络。

你可能感兴趣的:(神经网络,深度学习,机器学习)