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

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

程序媛一枚~

您的鼓励是我创作的最大动力。

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值