multi-target not supporte

原因：

input和labels的维度：

[32,751] [32,1]

labels多了一个维度

解决：

conf.ce_loss(thetas, labels.squeeze(1))

FocalLoss：

交叉熵损失函数CrossEntropyLoss可以支持多分类，并且不需要one-hot。

下面是源码：

class FocalLoss(nn.Module):def __init__(self, gamma=0):super(FocalLoss, self).__init__()self.gamma = gammaself.ce = torch.nn.CrossEntropyLoss()def forward(self, input, target):logp = self.ce(input, target)p = torch.exp(-logp)loss = (1 - p) ** self.gamma * logpreturn loss.mean()

multi-target not supporte相关推荐

如何将SAP Multi Target应用部署到SAP云平台的Cloud Foundry环境去
SHINA是SAP HANA Interactive Education的缩写,是一个demo应用,用于演示如何开发SAP HANA原生应用. 这个应用包含了sample数据以及HANA数据库表,vi ...
GNU Make chapter 2 —— Makefile 介绍
Makefile是由一系列的rule规则组成,这些rule都遵循以下形式: target ... : prerequisites ...command...... target(目标) 一般来说是需要 ...
Complexer-YOLO：基于语义点云的实时三维目标检测与跟踪
泡泡点云时空,带你精读点云领域顶级会议文章标题:Complexer-YOLO: Real-Time 3D Object Detection and Tracking on Semantic Poin ...
Linux引导过程与服务控制、排障演示及运行级别（详细图解）
Linux操作系统引导过程目录一.Linux操作系统引导过程 1.开机自检 2.MBR 引导 3.GRUB菜单 4.加载Linux内核 5.init进程初始化二.系统初始化进程init和Syste ...
详细图文演示——排除启动类故障以及Linux操作系统引导、运行级别和优化启动等相关知识
详细图文演示--排除启动类故障以及Linux操作系统引导.运行级别和优化启动等相关知识一.Linux操作系统引导过程 1.开机自检 2.MBR 引导 3.GRUB菜单 4.加载Linux内核 5.i ...
GAN domian adaptation
GAN 发展:domain 变多 DC GAN cycle GAN Combo GAN Star GAN DA: 基于分布对齐, 基于对抗等等一般:one source one target mu ...
计算机技术电子出版参考文献,参考文献规范
一.要求 1.在文后的参考文献表中,各条参考文献应按其在正文中出现的先后用阿拉伯数字连续排序.注意一定要按在文中出现的顺序编号. 2.文后参考文献表中的中文参考文献请改为中英文对照. 3.文后期刊类. ...
Unity-AR 简介
Unity-AR 简介现有Unity AR Sdk ARKit : 苹果推出的AR开发平台 ARCore : Google 推出的增强现实 SDK ARFoundation : ARFoundati ...
lua和php那个好,phpor
LuaPanda LuaPanda 是一个基于 VS Code 的 lua 代码工具,设计目标是简单易用.它支持多种开发框架,主要提供以下功能: 代码补全(code completion) 代码片段( ...
MRTG教程（五）：MRTG的配置参考指南
mrtg的配置参考指南内容 NAME OVERVIEW SYNTAX GLOBAL KEYWORDS OPTIONAL GLOBAL KEYWORDS EXAMPLES NAME mrtg-refe ...

multi-target not supporte

multi-target not supporte相关推荐

最新文章

热门文章