nll loss 和CrossEntropyLoss 的一些区别

最新推荐文章于 2023-10-25 16:08:55 发布

bald

最新推荐文章于 2023-10-25 16:08:55 发布

阅读量1.5k

点赞数

分类专栏：每天一个写bug小技巧 torch

本文链接：https://blog.csdn.net/Aries_M/article/details/111872772

版权

NLL Loss在传入这个loss前，需要先对输入进行一次 log_softmax 的变换, 例子如下：import torchimport torch.nn as nnimport torch.nn.functional as Fseed=1m = nn.LogSoftmax(dim=1)loss = nn.NLLLoss()# input is of size N x C = 3 x 5input = torch.randn(3, 5, requires_grad=True)pri

摘要由CSDN通过智能技术生成

NLL Loss

在传入这个loss前，需要先对输入进行一次 log_softmax 的变换, 例子如下：

import torch
import torch.nn as nn
import torch.nn.functional as F

seed=1
m = nn.LogSoftmax(dim=1)
loss = nn.NLLLoss()
# input is of size N x C = 3 x 5
input = torch.randn(3, 5, requires_grad=True)

最低0.47元/天解锁文章

确定要放弃本次机会？

福利倒计时

: :

立减 ¥

普通VIP年卡可用

立即使用

bald

关注关注

0
点赞
踩
0

收藏

觉得还不错? 一键收藏
0
评论
nll loss 和CrossEntropyLoss 的一些区别

NLL Loss在传入这个loss前，需要先对输入进行一次 log_softmax 的变换, 例子如下：import torchimport torch.nn as nnimport torch.nn.functional as Fseed=1m = nn.LogSoftmax(dim=1)loss = nn.NLLLoss()# input is of size N x C = 3 x 5input = torch.randn(3, 5, requires_grad=True)pri
复制链接

扫一扫