Pytorch训练网络时报的错:RuntimeError: cuda runtime error (59) : device-side assert triggered

        网络上普遍对于这个报错信息得出的问题为:标签分类错误、标签分类超限等等,我使用以下代码段对训练集和验证集进行脏数据搜索,发现数据没有问题。

        并且使用了Inria数据集与Potsdam数据集进行语义分割,发现同样会在几个epoch后出现这个错误,并且用了pycharm与spyder两个不同的编译器,错误仍然存在,因此错误原因大概率在代码中。

import os
import PIL.Image as Image
import numpy as np
import cv2

# 遍历文件夹
def walkFile(file):
    for root, dirs, files in os.walk(file):

        # root 表示当前正在访问的文件夹路径
        # dirs 表示该文件夹下的子目录名list
        # files 表示该文件夹下的文件list

        # 遍历文件
        for f in files:
            img = cv2.imread(os.path.join(root, f))
            for j in range(256):
                for k in range(256):
                    if (img[j][k][0] == 255 and img[j][k][1] == 0 and img[j][k][2] == 255) or  (img[j][k][0] == 0 and img[j][k][1] == 0 and img[j][k][2] == 0) or (img[j][k][0] == 0 and img[j][k][1] == 0 and img[j][k][2] == 255) or (img[j][k][0] == 255 and im
  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值