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

2. PyTorch简介:使用PyTorch训练第一个神经网络

2. PyTorch简介:使用PyTorch训练第一个神经网络

这篇博客将介绍如何使用PyTorch深度学习库训练第一个神经网络。具体来说是如何在scikit learn的make_blobs函数生成的示例数据集上训练PyTorch神经网络。

  1. 如何用PyTorch定义基本的神经网络结构
  2. 如何定义损失函数和优化器
  3. 如何正确地将梯度归零、执行反向传播(perform backpropagation)和更新模型参数-大多数新加入PyTorch的深度学习实践者在这一步中都会犯错误

PyTorch库初学者可能遇到的最大错误是忘记或混淆了以下步骤:

  • 将之前步骤中的梯度归零(opt.zero_grad())
  • 执行反向传播(loss.backward())
  • 更新模型参数(opt.step())
    在使用PyTorch时,如果不按照这个确切的顺序执行这些步骤,有搬起石头砸自己的脚的感觉,更糟糕的是,如果你混淆了这些步骤,PyTorch不会报告错误,因此你甚至可能不知道这块有错误!

PyTorch库功能强大,但使用PyTorch训练神经网络就像脱下自行车的训练轮——如果混淆了重要步骤,就没有安全网的保护了(不像Keras/TensorFlow,它允许将整个训练过程封装到一个model.fit调用中。
这并不是说Keras/TensorFlow比PyTorch“更好”,这只是两个深度学习库之间的区别。

1. 效果图

前几行输出显示了简单的4

相关文章:

  • LeetCode 295. 数据流的中位数(C++)*
  • MySQL表的增删查改
  • 完全二叉树与堆(包含STL堆的用法)
  • 用HTML写一个2023跨年动画代码(烟花+自定义文字+背景音乐+雪花+倒计时)
  • Electron打包图标不显示解决方案
  • Synchronized锁原理及 ConcurrentHashMap
  • OpenCV学习
  • 【位图】面对海量数据,如何压缩空间?定位数据?
  • 【Linux】进程创建、终止、等待、替换、shell派生子进程的理解…
  • 《L1 - 5分合集总结》
  • CSAPP Shell Lab
  • 13.0、Linux-安装Tomcat、以及防火墙常用命令
  • html实现酷炫的公司年会抽奖(附源码)
  • 自定义启动器
  • QT 学习笔记(十七)
  • 光敏二极管血氧仪方案设计研发
  • 【Kotlin】标准库函数 ③ ( with 标准库函数 | also 标准库函数 )
  • elementui el-table表格实现翻页和搜索均保持勾选状态(后端分页)
  • 内含JAVA简单概括和JAVA所需安装的软件和详细教程,想学习JAVA无从下手,这篇文章带你迈出第一步
  • C++——类和对象1
  • 屋顶上的阳光与火光:战争如何改变了加沙的能源格局
  • 六朝文物草连空——丹阳句容南朝石刻考察纪
  • 申花四连胜领跑中超联赛,下轮榜首大战对蓉城将是硬仗考验
  • 大学2025丨专访北邮校长徐坤:工科教育要真正回归工程本质
  • 持续更新丨伊朗内政部长:港口爆炸已致8人死亡750人受伤
  • 中国人民银行行长潘功胜会见世界银行行长彭安杰