光说不练假把式,现在我们已经积累了那么多的 PyTorch 知识,让我们实践一下吧!
本文从简单的手写数字识别入手,参考了若干文章:
- Handwritten Digit Recognition Using PyTorch — Intro To Neural Networks
- Building Your First PyTorch Solution
1. PyTotch 使用总览
使用 PyTorch 进行深度学习的步骤主要分以下七步:
- 准备数据,包括数据的预处理和封装;
- 模型搭建;
- 选择损失函数;
- 选择优化器;
- 迭代训练;
- 评估模型;
- 保存模型。