一、问题贴图和描述
numpy.core._exceptions.MemoryError: Unable to allocate 128. KiB for an array with shape (16384,) and data type float64
二、问题产生原因和解决方法
原因一:一般是由于要跑的数据集太大,而自己电脑没有那么大的内存去支持它去把这个模型跑起来。
原因二:batch_size调到太大
解决方法:对数据集进行采样,只采用其中的一部分数据,同时调小batch_size,增加训练的轮数(epochs)