
pytorch loss = loss_func(output, label) 报错
发布日期:2021-05-06 23:49:09
浏览次数:10
分类:技术文章
本文共 424 字,大约阅读时间需要 1 分钟。
在运行损失函数时 loss = loss_func(output, label) 时报了两个错 :
1.
RuntimeError: Expected object of scalar type Long but got scalar type Float for argument #2 'target'
这是因为label必须是 LongTensor 类型, 之前是 :
label = t.tensor(float(image[1])) # image[1]为str类型
改为 :
label = t.tensor(float(image[1])).long()
2.
Assertion `cur_target >= 0 && cur_target < n_classes’ failed
这是因为我的数据集类别是从1开始,需要依次减一,所以要改为 :
label = t.tensor(float(image[1])-1).long()
发表评论
最新留言
表示我来过!
[***.240.166.169]2025年03月31日 14时49分36秒
关于作者

喝酒易醉,品茶养心,人生如梦,品茶悟道,何以解忧?唯有杜康!
-- 愿君每日到此一游!
推荐文章
IDEA 成功在tomcat上部署项目
2019-03-04
Node.js response 页面中文乱码
2019-03-04
谷歌浏览器 禁用JavaScript
2019-03-04
gitee 修改个人域名 个人空间地址 URL
2019-03-04
C++11中bind的使用错误
2019-03-04
Android中CMake的使用之一初步总结
2019-03-04
一起学智能合约之四表达式和控制结构
2019-03-04
futex同步机制分析之三内核实现
2019-03-04
多线程的伪共享
2019-03-04
flink分析使用之五工作图的生成和分发
2019-03-04
基于OpenCV的路面质量检测
2019-03-04
Spring Cloud系列_11 Feign负载均衡、请求传参
2019-03-04
leetcode 543. Diameter of Binary Tree
2019-03-04
VSLAM系列原创01讲 | 深入理解ORB关键点提取:原理+代码
2019-03-04
卡尔曼滤波器的特殊案例
2019-03-04
基于Opencv的图像单应性转换实战
2019-03-04
【C++简明教程】Python和C++指定元素排序比较
2019-03-04
视觉实战|使用人工神经网络进行图像分类
2019-03-04
3D感知技术及实践
2019-03-04
北大读博手记:怎样完成自己的博士生涯?非常具有指导性!
2019-03-04