RuntimeError: Found dtype Long but expected Float
今天在跑模型的时候，发现了以下的报错
RuntimeError: Found dtype Long but expected Float
这里的报错发生在BCELoss的部分
loss_fn = torch.nn.BCELoss()
logit = logit.view(logit.size()[0]*logit.size()[1],-1)
batch_label = batch_label.view(batch_label.size()[0]*batch_label.size()[1],-1)
crossentropyloss = loss_fn(logit,batch_label)
这里的batch_label类型为torch.float，而crossentropyloss要求batch_label必须为torch.int类型
因此这段代码需要修改一下：
loss_fn = torch.nn.BCELoss()
logit = logit.view(logit.size()[0]*logit.size()[1],-1)
batch_label = batch_label.view(batch_label.size()[0]*batch_label.size()[1],-1)
batch_label = batch_label.to(torch.float)
crossentropyloss = loss_fn(logit,batch_label)
RuntimeError: Found dtype Long but expected Float相关推荐
- RuntimeError: Found dtype Double but expected Float”
RuntimeError: Found dtype Double but expected Float" 在求损失函数时报错了, 解决方法: target.float() a=np.arra ...
- PyTorch出现错误“RuntimeError: Found dtype Double but expected Float”
1 错误描述 今天在调试PyTorch代码时出现"RuntimeError: Found dtype Double but expected Float"的错误,相关提示信息如下 ...
- python将object转换为float_将pypeas.Series从dtype对象转换为float，将错误转换为nans
如何将pandas.Series从dtype转换object为float,并将错误转换为nans ? 自v0.17起,convert_objects不推荐使用. 为了一系列转换为数字,使用pd.to_ ...
- python导出csv 字符串变成数字_python – pandas read_csv列dtype设置为十进制但转换为字符串...
我正在使用pandas(v0.18.1)从名为'test.csv'的文件中导入以下数据: a,b,c,d 1,1,1,1.0 我已经为列'c'和'd'将dtype设置为'decimal.Decimal ...
- kaldi debug：Failed to read token [started at file position -1], expected IvectorExtractorStats
Kaldi Debug:ERROR (ivector-extractor-sum-accs[5.2]:ExpectToken():io-funcs.cc:197) Failed to read tok ...
- NumPy之:数据类型对象dtype
文章目录 简介 dtype的定义 可转换为dtype的对象 dtype对象 None 数组标量类型 通用类型 内置Python类型 带有.dtype属性的对象 一个字符的string对象 数组类型的S ...
- python dtype强制类型转换_Numpy数据类型转换astype,dtype的方法
1.查看数据类型 In [11]: arr = np.array([1,2,3,4,5]) In [12]: arr Out[12]: array([1, 2, 3, 4, 5]) // 该命令查看数 ...
- 3. Numpy中的数据类型对象 ( dtype)
<玩转Numpy计算库>视频课程 <玩转Numpy计算库>视频课程链接:https://edu.csdn.net/course/detail/28656 数据类型对象 ( dt ...
- Numpy数据类型转换astype，dtype
1.查看数据类型 In [11]: arr = np.array([1,2,3,4,5])In [12]: arr Out[12]: array([1, 2, 3, 4, 5])// 该命令查看数据类 ...
最新文章
- 102页PPT，DeepMind强化学习最新进展，含图文、公式和代码
- Java并发—锁的四种状态
- 初学web前端开发，你必要了解的基本概念与工具技术
- 笔记15（Linux监控平台）
- Tomcat性能调优及JVM内存工作原理
- LinkedHashMap内部实现
- 每天10个Linux命令四
- linux 信号阻塞和信号未决
- Linux视频编解码库，Ubuntu 18.04 FFMPEG最新版本安装总结
- Android五子棋游戏源码详解
- c语言中如何找坐标,如何在C语言中找到四个坐标并擦除RubberBand矩形的绘图#
- LiveData更新数据报错java.lang.IllegalStateException: Cannot invoke setValue on a background thread
- 【Python】爬虫爬取各大网站新闻（一）
- RGB颜色与颜色名称对照表
- 手把手教你VMware虚拟机详细安装Debian 11 图文教程
- STM32CubeMX的使用教程
- 专利第三方评估_什么是第三方风险评估，您该如何做？
- STM32 - ADC采集电压 中断处理
- Java学习笔记【入门】（1）：Java版本介绍
- IT审计 独立于IT管理的监督过程(zt)