当前位置： 首页 > news >正文

nn.CrossEntropyLoss的ignore_index标签(CE loss)

news 来源：原创 2024/5/16 9:16:31

 例如我的pred是(b,2,w,h),而label索引是(b,1,w,h)的矩阵，其中只有0，1值，0值代表从pred的第0个通道选择像素值，1值代表从pred的第1个通道选择像素值。

而此时我发现因为程序的错误，label矩阵中混入了一些-1值，这样正常的话是会报错的，因为pred矩阵没有-1通道。此时最简单的一个方法就是

loss = nn.CrossEntropyLoss(ignore_index=-1)

就是相当于忽略-1标签

值为-1的位置的对应像素值就不参与计算梯度了

相关文章：

  • 装系统时的UEFI模式
  • Pytorch dataloader中的num_workers (选择最合适的num_workers值)
  • rsync数据传输
  • Pytorch torch.utils.data.DataLoader(二) —— pin_memory锁页内存 drop_last num_works
  • Pytorch分布式训练/多卡训练(三) —— Model Parallel 并行
  • Pytorch backend 通信后端
  • PyTorch多卡/多GPU/分布式DPP的基本概念(noderanklocal_ranknnodesnode_ranknproc_per_nodeworld_size)
  • Pytorch函数keepdim=True
  • Python opencv putText()中文乱码问题
  • Python类的__call__方法
  • Python处理XML(ElementTree)
  • Pytorch为不同层设置不同的学习率(全局微调)
  • PIL Image和opencv读入图片相互转化
  • Python 星号表达式*(starred expression / unpack / 解包)
  • Pytorch/Python计算交并比IOU(IU)代码(批量算IOU)
  • 【面试系列】之二：关于js原型
  • Android交互
  • express如何解决request entity too large问题
  • Java读取Properties文件的六种方法
  • Making An Indicator With Pure CSS
  • ubuntu 下nginx安装 并支持https协议
  • 彻底搞懂浏览器Event-loop
  • 等保2.0 | 几维安全发布等保检测、等保加固专版 加速企业等保合规
  • 精益 React 学习指南 （Lean React）- 1.5 React 与 DOM
  • 快速体验 Sentinel 集群限流功能，只需简单几步
  • 软件开发学习的5大技巧，你知道吗？
  • 使用 Docker 部署 Spring Boot项目
  • 为视图添加丝滑的水波纹
  • 问题之ssh中Host key verification failed的解决
  • 学习Vue.js的五个小例子
  • 终端用户监控：真实用户监控还是模拟监控？
  • RDS-Mysql 物理备份恢复到本地数据库上
  • 策略 ： 一文教你成为人工智能（AI）领域专家
  • #微信小程序（布局、渲染层基础知识）
  • (1/2) 为了理解 UWP 的启动流程，我从零开始创建了一个 UWP 程序
  • （3）STL算法之搜索
  • （超简单）构建高可用网络应用：使用Nginx进行负载均衡与健康检查
  • （待修改）PyG安装步骤
  • （附源码）spring boot基于Java的电影院售票与管理系统毕业设计 011449
  • （六）什么是Vite——热更新时vite、webpack做了什么
  • （十八）用JAVA编写MP3解码器——迷你播放器
  • （一） springboot详细介绍
  • (原創) 如何解决make kernel时『clock skew detected』的warning? (OS) (Linux)
  • （转）关于如何学好游戏3D引擎编程的一些经验
  • (转)使用VMware vSphere标准交换机设置网络连接
  • (自适应手机端)响应式新闻博客知识类pbootcms网站模板 自媒体运营博客网站源码下载
  • .java 9 找不到符号_java找不到符号
  • .NET Core 网络数据采集 -- 使用AngleSharp做html解析
  • .NET Core工程编译事件$(TargetDir)变量为空引发的思考
  • .net 前台table如何加一列下拉框_如何用Word编辑参考文献
  • .net与java建立WebService再互相调用
  • 。Net下Windows服务程序开发疑惑
  • /proc/stat文件详解（翻译）
  • @Bean有哪些属性
  • [ C++ ] STL---仿函数与priority_queue