pytorch criterion踩坑小结

PyTorch中CrossEntropyLoss使用详解
最新推荐文章于 2025-10-10 16:56:00 发布
原创 最新推荐文章于 2025-10-10 16:56:00 发布 · 3.1w 阅读 · 51 ·
CC 4.0 BY-SA版权
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
文章标签：

#pytorch #python

1. 数据类型不匹配：

报错：Expected object of type torch.LongTensor but found type torch.FloatTensor for argument #2 ‘target’

criterion = nn.CrossEntropyLoss()
loss = criterion(y_pre, y_train)

这里的y_train类型一定要是LongTensor的，所以在写DataSet的时候返回的label就要是LongTensor类型的，如下

def__init__(self, ...):
	self<
最低0.47元/天 解锁文章
决策树概述+模块介绍+重要参数（criterion+random_state&splitter+减枝参数+目标权重参数）+回归树（参数+实例+拟合正弦曲线）+泰坦尼克号生存者预测实例
weixin_51589123的博客
09-21 7065
文章目录什么是sklearn一、决策树概述（一）概述（二）基础概念（三）决策树算法的核心是要解决两个问题：二、模块sklearn.tree的使用（一） 模块介绍（二）使用介绍三、重要参数（一）criterion（二） random_state & splitter（三）减枝参数1、max_depth2、min_samples_leaf & min_samples_split min_samples_leaf3、max_features & min_impurity_decrease（
决策树③——决策树参数介绍（分类和回归）
机器学习初学者必看，关注我，一起了解机器学习
02-22 5613
前面总结了信息熵，信息增益和基尼信息的定义，以及决策树三大算法的原理及迭代过程，今天介绍下Python中机器学习Sklearn库中决策树的使用参数 决策树既可以做分类，也可以做回归，两者参数大体相近，下面会先介绍分类，再对回归不一样的参数做单独说明 一、分类参数 1、 criterion: 特征选取方法，可以是gini（基尼系数），entropy（信息增益），通常选择gini，即CART算法，如果...
Python：None 类型
最新发布
qq_41176800的博客
10-10 380
（3）在数值计算中，应使用 math.nan 或 numpy.nan 表示无效数值，以避免类型错误并支持数学函数识别。在比较时推荐使用 is / is not，并注意 None 与 0、空字符串、False 并不等价。在数据结构中，None 常用来表示“缺失值”或“未赋值状态”，例如列表或字典中的占位符。是一种特殊的单例类型，通常用于占位、函数返回值缺省标记，以及逻辑判断。（1）None 不是 0、空字符串或 False，它是一个独立的类型。，用于表示“无值”或“未定义的状态”，用于表示“无”或“缺失”。
神经网络图文检索模型预测代码学习总结（二）
qq_52648919的博客
08-03 2732
本文主要用于积累自己学习过程中搭建神经网络的常见代码，如有不准确之处，欢迎各路大神指出！谢谢！
pytorch 损失函数(平方损失、二分类损失、交叉熵损失等)
彭伟的博客
11-22 1万+
　损失函数 损失函数，又叫目标函数，是编译一个神经网络模型必须的两个参数之一。另一个必不可少的参数是优化器。 损失函数是指用于计算标签值和预测值之间差异的函数，在机器学习过程中，有多种损失函数可供选择，典型的有距离向量，绝对值向量等。 上图是一个用来模拟线性方程自动学习的示意图。粗线是真实的线性方程，虚线是迭代过程的示意，w1 是第一次迭代的权重，w2 是第二次迭代的权重，w3 是第三次迭代的权...
Pytorch学习之十九种损失函数
热门推荐
mingo_敏
01-01 7万+
损失函数通过torch.nn包实现， 1 基本用法 criterion = LossCriterion() #构造函数有自己的参数 loss = criterion(x, y) #调用标准时也有参数 2 损失函数 2-1 L1范数损失 L1Loss torch.nn.L1Loss(size_average=True) 2-2 均方误差损失 MSELoss torch.nn.MSELoss(siz...
Pytorch安装踩坑
qq_46003434的博客
07-31 1302
个人学习时踩坑的经验，仅作为记录。
windows环境下pytorch安装踩坑
AIGIS的博客
12-21 1799
这两天由于要使用Genesis，需要用到pytorch的环境，就装了一下pytorch，记录一下过程和踩坑记录。因为电脑上要使用ArcGIS，因此默认的python环境是不能改的，否则ArcGIS就不能启动了。因此新的python的环境需要使用AnacondaCUDA: 12.4Anaconda：2022.10Python：3.9Pytorch本文详细的讲解了Pytorch的安装过程，以及中间遇到各种坑，最后总结出来一个最佳实践，希望对读者有所帮助，回见~
Ubuntu20.04安装pytorch3d踩坑记录
tanking66的博客
03-07 1981
显卡：RTX4070Tipytorch3d官方发布链接：https://github.com/facebookresearch/pytorch3d/blob/main/INSTALL.md多看官方安装教程。
Pytorch Lightning 踩坑记录
jaffe的博客
07-07 1999
最近两周开始了解和上手学习pytorch lightning，这个框架，csdn和知乎资料都比较少，而且框架相对年轻，还是有不少该改进的地方 lightning 可以和torch 兼容，在某些方面反而有冲突，可能需要自己写callback 函数实现 今天记录一下自己的踩坑(不分先后) 模型test 无输出 正确代码 def test_step(self, batch, batch_idx): x, y = batch logits = self.forward(x) preds =
pytorch的基本使用
yanhe156的博客
10-05 2536
需要特别注意的一点 很多博客中提到了pytorch中的tensor基本上和numpy中的array是一样的，可以直接通过函数相互转化，同时pytorch中有variable变量，variable是对tensor的一个封装，用于构建计算图，并进行迭代。但是对0.4.1版本来说，Variable不再被支持使用。 可以理解为Variable合并到了Tensor中。 我开始一些博客和文档同时看，一直以为t...
pytorch简单的二分类
weixin_38241876的博客
12-28 1413
train.py # from torchvision.models.resnet import resnet34 # from dataset import datasest # from torch.optim.lr_scheduler import StepLR # from torch.nn import CrossEntropyLoss # import torch # from torch.utils.tensorboard import SummaryWriter # # train_ro
PyTorch经验指南：技巧与陷阱_def one_iteration(model, data, target, criterion)
m0_61067876的博客
05-12 876
【代码】PyTorch经验指南：技巧与陷阱_def one_iteration(model, data, target, criterion)
【pytorch】——报错/pytorch/aten/src/THCUNN/ClassNLLCriterion.cu:106: void cunn_ClassNLLCriterion_updateOu
怡宝2号
09-11 2697
转载自：https://blog.youkuaiyun.com/littlehaes/article/details/102806323 头一次使用pytorch, 调用交叉熵损失函数nn.CrossEntropyLoss()的时候报错 /pytorch/aten/src/THCUNN/ClassNLLCriterion.cu:106: void cunn_ClassNLLCriterion_updateOutput_kernel(Dtype *, Dtype *, Dtype *, long *, Dtype *,
PyTorch项目应用实例(八)固定权重|顺序训练网络
祥瑞的技术博客
05-28 3501
背景：需要将模型分层训练，不能同时训练。即固定一部分权重训练另一部分。 目录 一、多输出网络 1.1 heads多输出 1.2 最终的输出 二、loss更新网络的方法 2.1 loss位置 2.2 criterion定义 2.3 optimizer定义 2.4 定义顺序汇总 三、更改相应代码 3.1 定义需要优化的参数 3.2 更改学习率 3.3 网络loss及结构 一...
[torch]criterion/maskzero
apsvvfb的专栏
05-06 772
require 'nn' require 'rnn'maxzero = 1 if maxzero == 1 then y_ = torch.Tensor{{1,5,2},{0,5,2},{0,0,0}} y = torch.Tensor{{2,3,1},{0,5,4},{0,0,0}} y = torch.Tensor{{2,3,1},{0,5,4},
Pytorch的四种函数讲解
shiguang521314的博客
05-16 1051
在PyTorch中，Softmax和LogSoftmax函数是用于将多维张量的值转换为概率分布的函数。它们通常用于多分类问题中，将模型的输出转换为每个类别的概率。在PyTorch中，NLLLoss（负对数似然损失）和（交叉熵损失）是两种常用的损失函数，它们通常用于多分类问题。一、四种函数的概念LogSoftmax:LogSoftmax函数是Softmax函数的对数形式，它将输入的任意实数值映射到(0, 1)区间，并且输出的总和为1。
PyTorch学习之：深入理解神经网络
www_tlj的专栏
04-02 1822
首先，你需要通过继承nn.Module类来定义你自己的网络结构。在你的类中，你将初始化网络层并定义前向传播路径。self.fc1 = nn.Linear(input_size, hidden_size) # 第一个全连接层self.relu = nn.ReLU() # 激活函数self.fc2 = nn.Linear(hidden_size, num_classes) # 第二个全连接层return out在这个例子中，__init__方法用于初始化网络中的各个层。forward。
PyTorch学习（5）:损失函数
小花生的博客
06-27 3933
PyTorch学习（5）:损失函数 Pytorch官方文档： https://pytorch-cn.readthedocs.io/zh/latest/ Pytorch学习文档： https://github.com/tensor-yu/PyTorch_Tutorial 参考： https://pytorch.org/docs/stable/nn.html     https://pytorch-cn.readthedocs.io/zh/latest/package_references/torch-nn/
评论 7
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值