ResourceExhaustedError: OOM when allocating tensor with shape
在使用keras训练模型时报错,问题原因:
GPU显存不足,解决方案有:
1. 终端使用nvidia-smi命令查看显卡状态,将其他占用进程kill掉。
2. 将训练的batch_size调小
MemoryError
在将数据归一化时报错,问题原因:
数据size过大((10000,224,224,3)),归一化时需要转化成浮点数,每个数据占用字节16,共需要内存16x10000x224x224x3
解决方案:
分批次导入数据
EOFError: Ran out of input
pickle.load(filename) 导入数据报错。
在终端使用
ls -l|grep "filename"
查看后发现该文件是空的,故报错。