RuntimeError Assertion cur_target = 0 cur_target n

RuntimeError: Assertion `cur_target >= 0 && cur_target < n_classes' failed

criterion = torch.nn.CrossEntropyLoss().cuda()
output=output.cuda()
target=target.cuda()
loss=criterion(output, target)

但我终于找到了我的错误：输出就像tensor([[0.5746,0.4254]])和目标一样tensor([2])，数字2超出了输出的指示

当我不使用GPU时，此错误消息是：

RuntimeError: Assertion `cur_target >= 0 && cur_target < n_classes' failed.  at /opt/conda/conda-bld/pytorch-nightly_1547458468907/work/aten/src/THNN/generic/ClassNLLCriterion.c:93

RuntimeError Assertion cur_target = 0 cur_target n_classes failed相关推荐

MaskRCNN-Benchmark框架Assertion 't ** 0 t ** n_classes' failed可能的原因
Mask R-CNN Benchmark是一个完全由PyTorch 1.0写成,快速.模块化的Faster R-CNN和Mask R-CNN组件.该项目旨在让用户更容易地创建一种模块,实现对图片中物品 ...
RuntimeError: Assertion cur_target 0 cur_target n_classes failed
问题描述使用pytorch的函数 torch.nn.CrossEntropyLoss()计算Loss时报错: RuntimeError: Assertion `cur_target >= 0 ...
PyTorch使用F.cross_entropy报错Assertion `t ＞= 0 t ＜ n_classes` failed问题记录
前言在PyTorch框架下使用F.cross_entropy()函数时,偶尔会报错ClassNLLCriterion ··· Assertion `t >= 0 && t &l ...
CvSVM::EPS_SVR train_auto assertion sv_count != 0 failed原因
CvSVM::EPS_SVR train_auto assertion sv_count != 0 failed原因: 出现这种情况,一般表示你的训练集和结果数据集,没有输入到你的Mat结构中,如果用 ...
问题解决：shared_ptr Assertion px != 0 failed 及debug经验分享
问题解决:shared_ptr Assertion px != 0 failed及debug经验分享问题详细描述: /usr/include/boost/smart_ptr/shared_ptr.h ...
assertion: 18 { ok: 0.0, errmsg: auth failed, code: 18 }
0.mongorestore报错 mongorestore -h 172.0.0.10:xxx -u xxx -p xxx -d aeras 执行恢复数据库出现RT鉴权失败错误 assertion: ...
sphinx error connection to 127.0.0.1:9312 failed (errno=0, msg=)
通过一键安装lnmp环境,运行sphinx时,提示sphinx error connection to 127.0.0.1:9312 failed (errno=0, msg=) 是因为lnmpa服务 ...
ERROR in ./node_modules/element-ui/lib/theme-chalk/fonts/element-icons.ttf 1:0 Module parse failed:
在使用 webpack 搭建 vue 项目时,添加了element-ui 在启动 webpack-dev-server 后报错 ERROR in ./node_modules/element-ui/l ...
成功解决 raise RuntimeError(“The JPMML-SkLearn conversion application has failed. The Java executable
成功解决 raise RuntimeError("The JPMML-SkLearn conversion application has failed. The Java execu ...

RuntimeError Assertion cur_target = 0 cur_target n_classes failed

RuntimeError Assertion cur_target = 0 cur_target n_classes failed相关推荐

最新文章

热门文章