RuntimeError: CUDA error: device-side assert triggered

问题:

我发现自己densenet网络就报错:

RuntimeError: CUDA error: device-side assert triggered

原因:

在给出label的时候,我的数据集在某个标签上会给进去一个小于零或者大于类别数的一个标签(在我这边是大于类别数了,本来在训练集3个类别,我程序也接收3个类别,但是现在训练集变成4个类别,而程序依然只接收3个类别,所以出错)。
输出:

 testset = torchvision.datasets.ImageFolder(root='/home/zhongjia/Spaceweather/primary/continuum_data/continuum_new/test', transform=transform_test)#634+1987+400=3021
    print(testset.classes)
    print(testset.class_to_idx)
    print(testset.imgs)

可以看到类别数存在id为3的图片,而我只有0,1,2共3类图片

可以看到现在因为多了.ipynb_checkpoints,有4个分类,把.ipynb_checkpoints删除即可 

  • 2
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值