江苏龙网

  1. 首页 > 人文 > >

PyTorch，这款深度学习框架你值得拥有，所以在这里( 二 )


数据
接下来是数据的准备部分 ， 代码也是完全相同的 ， 只不过Lightning做了这样的处理 。
它将PyTorch代码组织成了4个函数 ， prepare_data、train_dataloader、val_dataloader、test_dataloader
PyTorch，这款深度学习框架你值得拥有，所以在这里
文章图片
prepare_data
这样就解决了PyTorch老是重复处理数据的问题 ， 这样速度也就提上来了 。
【PyTorch，这款深度学习框架你值得拥有，所以在这里】train_dataloader ，val_dataloader ，test_dataloader
每一个都负责返回相应的数据分割 ， 这样就能很清楚的知道数据是如何作的 ， 在以往的教程里 ， 都几乎看不到它们的是如何操作数据的 。
此外 ， Lightning还允许使用多个dataloaders来或验证 。
优化
接着就是优化 。
PyTorch，这款深度学习框架你值得拥有，所以在这里
文章图片
不同的是 ， Lightning被组织到配置优化器的功能中 。 如果你想要使用多个优化器 ， 则可同时返回两者 。
PyTorch，这款深度学习框架你值得拥有，所以在这里
文章图片
损失函数
对于n向分类 ， 我们要计算交叉熵损失 。 两者的代码是完全一样的 。
PyTorch，这款深度学习框架你值得拥有，所以在这里
文章图片
此外 ， 还有更为直观的—验证和训练循环 。
PyTorch，这款深度学习框架你值得拥有，所以在这里
文章图片
在PyTorch中 ， 我们知道 ， 需要你自己去构建for循环 ， 可能简单的项目还好 ， 但是一遇到更加复杂高级的项目就很容易翻车了 。
而Lightning里这些抽象化的代码 ， 其背后就是由Lightning里强大的trainer团队负责了 。
PyTorch Lightning安装教程
看到这里 ， 是不是也想安装下来试一试 。
PyTorch Lightning安装十分简单 。
代码如下：
condaactivate my_env
pip install pytorch-lightning
或在没有conda环境的情况下 ， 可以在任何地方使用pip 。
代码如下：
pipinstall pytorch-lightning
创建者也有大来头
2018年 ， 从哥伦比亚大学计算机科学与统计学专业毕业 ， 本科期间 ， 他还曾辅修数学 。
现在已获得Google Deepmind资助攻读博士学位的奖学金 ， 去年还收到Facebook AI Research实习邀请 。
此外 ， 他还曾是一个海军军官 ， 接受过美国海军海豹突击队的训练 。
PyTorch，这款深度学习框架你值得拥有，所以在这里
文章图片
前不久 ，华尔街日报就曾还曾提到这个团队 ， 他们正在研究呼吸疾病与呼吸模式之间的 。 可能会应用到的场景 ， 是通过电话在诊断新冠症状 。 目前 ， 该团队还处在数据收集阶段 。
果然 ， 优秀的人 ， 干什么都是优秀的 。 叹气…
上手传送门
创建者个人网站：
完—


推荐阅读


上一篇：与产品经理，以及为什么产品经理要学商业分析

下一篇：私生饭|陈雅婷的“透视装”有多美？难怪王思聪会这么喜欢，网友：这身材心头肉