0
点赞
收藏
分享

微信扫一扫

3.内存溢出错误

一、问题贴图和描述

numpy.core._exceptions.MemoryError: Unable to allocate 128. KiB for an array with shape (16384,) and data type float64
在这里插入图片描述

二、问题产生原因和解决方法

原因一:一般是由于要跑的数据集太大,而自己电脑没有那么大的内存去支持它去把这个模型跑起来。
原因二:batch_size调到太大

解决方法:对数据集进行采样,只采用其中的一部分数据,同时调小batch_size,增加训练的轮数(epochs)

举报

相关推荐

0 条评论