2.1.3 构建模型

使用PyTorch实现神经网络的关键就是选择网络层，构建模型，然后选择损失函数和优化器。在nn工具箱中，可以直接引用的网络很多，有全连接层、卷积层、循环层、归一化层、激活层等。

PyTorch构建模型大致有以下两种方式。

1）继承nn.Module基类来构建模型。

2）使用PyTorch Lightning来构建模型。