4.6 PyTorch使用损失函数

本文是《PyTorch深度学习实践》专栏的一篇内容,介绍了PyTorch中适用于回归和分类的损失函数,包括mse_loss和l1损失函数,以及分类常用的损失函数,讲解了reduction参数的配置选项。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

欢迎订阅本专栏:《PyTorch深度学习实践》
订阅地址:https://blog.youkuaiyun.com/sinat_33761963/category_9720080.html

  • 第二章:认识Tensor的类型、创建、存储、api等,打好Tensor的基础,是进行PyTorch深度学习实践的重中之重的基础。
  • 第三章:学习PyTorch如何读入各种外部数据
  • 第四章:利用PyTorch从头到尾创建、训练、评估一个模型,理解与熟悉PyTorch实现模型的每个步骤,用到的模块与方法。
  • 第五章:学习如何利用PyTorch提供的3种方法去创建各种模型结构。
  • 第六章:利用PyTorch实现简单与经典的模型全过程:简单二分类、手写字体识别、词向量的实现、自编码器实现。
  • 第七章利用PyTorch实现复杂模型:翻译机(nlp领域)、生成对抗网络(GAN)、强化学习(RL)、风格迁移(cv领域)。
  • 第八章:PyTorch的其他高级用法:模型在不同框架之间的迁移、可视化、多个GPU并行计算。

PyTorch提供了许多常用的损失函数,再前面小节中也有用到过了。损失函数可以在torch.nn.functional中调用。

再日常模型训练中,最常见的是做回归预测或者做分类,下面分别介绍它们的常用损失函数的用法。

其他损失函数的用法是一样的。可以再这份博客里看看各类损失的含义:https://www.cnblogs.com/kk17/p/10246324.html

4.6.1 适用于回归的损失函数

  • mse_loss 误差的平方
  • l1 误差的绝对值

注意可以通过reduction这个参数配置是均值、求和还是不进行聚合。

import numpy as np
import torch
import torch.nn.
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值