Tensorflow刚开始训练killed bug解决方案

最新推荐文章于 2024-08-23 08:49:09 发布

原创最新推荐文章于 2024-08-23 08:49:09 发布 · 5.4k 阅读

2 ·

CC 4.0 BY-SA版权

python 专栏收录该内容

13 篇文章

订阅专栏

本文分享了一个关于内存溢出的问题解决过程。作者在训练模型时遭遇内存不足，通过对比实验发现训练图片尺寸过大导致内存暴涨。通过将图片尺寸从5472×3648调整为450×300，成功解决了内存溢出问题，实现了模型的正常训练。

这个BUG弄了整整两天，记录一下
在这里插入图片描述
各种百度，各种换模型，搞得头昏脑胀，最后通过对比实验，监测到killed之前内存暴涨，怀疑是内存泄漏或者内存不够，原来的训练图片尺寸5472×3648,resize一下就可以正常训练了，resize代码如下

import os
from PIL import Image
 
filename = os.listdir("/home/wxy/下载/VOCdevkit/VOC2007/JPEGImages/")
base_dir = "/home/wxy/下载/VOCdevkit/VOC2007/JPEGImages/"
new_dir  = "/home/wxy/下载/VOCdevkit/VOC2007/JPEGImages919/"
size_m = 450
size_n = 300
 
for img in filename:
    image = Image.open(base_dir + img)
    image_size = image.resize((size_m, size_n),Image.ANTIALIAS)
    image_size.save(new_dir+ img)