Pytorch入门学习笔记（一）

本文结合小土堆的教学视频PyTorch深度学习快速入门教程进行学习

一、加载数据集

1.  加载数据集所需要的两个类 —— Dataset & DataLoader

Dataset：是用来对数据打包的抽象类，提供一种方式来获取数据及其label（进行数据分类和编号）。能够实现如何获取数据集中每一个数据及其label，并且告诉我们总共数据集中有多少个数据。

DataLoader：是用来处理模型输入数据的工具类，为网络提供不同的数据形式。

2. Dataset类的重写及加载数据集

 具体代码实现如下