详细内容或原文请订阅后点击阅览
使用 torch 对图像进行分类
我们了解迁移学习、输入管道和学习率调度程序,同时使用 torch 来区分美丽鸟类的种类。
来源:RStudio AI博客在最近的文章中,我们一直在探索 Torch 的基本功能:张量,每个深度学习框架的必要条件;autograd,Torch 的反向模式自动微分实现;模块,神经网络的可组合构建块;优化器,Torch 提供的优化算法。
torch
张量
autograd
torch
模块
优化器
torch
但我们还没有真正拥有“hello world”的时刻,至少如果你所说的“hello world”是指对宠物进行分类的不可避免的深度学习体验的话。猫还是狗?比格犬还是拳师犬?奇努克犬还是吉娃娃?我们将通过问一个(略微)不同的问题来区分自己:什么鸟?
宠物分类的深度学习经验我们将在途中讨论的主题:
- torch 数据集和数据加载器各自的核心角色。如何应用变换,用于图像预处理和数据增强。如何使用 Resnet(He et al. 2015),一个带有 torchvision 的预训练模型,进行迁移学习。如何使用学习率调度程序,特别是单周期学习率算法 [@abs-1708-07120]。如何找到一个好的初始学习率。
torch 数据集和数据加载器各自的核心角色。
torch
数据集
数据加载器
如何应用变换,用于图像预处理和数据增强。
变换
如何使用 Resnet(He et al. 2015),一个带有 torchvision 的预训练模型,进行迁移学习。
(He et al. 2015) He et al. 2015torchvision
如何使用学习率调度器,特别是单周期学习率算法 [@abs-1708-07120]。