Check failed: status == CUBLAS_STATUS_SUCCESS (11 vs. 0) CUBLAS_STATUS_MAPPING_ERROR

    caffe出现这个问题时,总让人摸不着头脑,不知从哪儿开始排查问题。
    网上搜索资料,发现有人说是label做得不对,比如只有1没有0。但是这个是问题吗?就不能只有一个标签吗?
    不过这种说法是说对了一半。
    我是做图像分割时出现这个问题的,caffe用opencv读取图像时,是读取的0–255,这样读取的label也是如此。这下就明白了,caffe如果读到了255,它就认为是有0–255的标签,但是我们最后的输出比这个维度小,这样就不对了,所以需要在输入图片是指定 scale

  transform_param {
    scale: 0.00390625
  }

总结:
制作标签时一定要注意从0开始一个个地往后排,不能弄错。

评论 13
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值