pytorch构建一个神经网络的步骤

定义数据类

• 数据集若不存在，则从预设网址下载数据到给定位置。并做数据预处理。
• 数据集若存在，做数据预处理。

定义网络层类

网络层可以有多种，因此可以有多个不同的网络层类。当定义框架类用到这个图层的时候，例如卷积层。就可以直接在框架类的构造函数里初始化一个网络层类的对象。

定义网络框架类

_init_函数，forward函数

定义train和test函数

train函数用于网络训练，test函数用于网络测试。
注意：要记录训练和测试的结果，并写入指定文件和出图。