我们需要使用torch.nn包中的工具来构建神经网络
首先明确构建一个神经网络需要以下几步:
- 定义神经网络的权重,搭建网络结构
- 遍历整个数据集进行训练遍历整个数据集进行训练
将数据输入神经网络
计算loss
计算网络权重的梯度
更新网络权重
weight = weight + learning_rate * gradient
nn.Conv2d的功能是:对由多个输入平面组成的输入信号进行二维卷积
输入信号的形式为(N,Cin,H,W)
N表示batch size,Cin表示channel个数,H ,W 分别表示特征图的高和宽。
参数说明: