使用PyTorch实现神经网络的关键就是选择网络层,构建模型,然后选择损失函数和优化器。在nn工具箱中,可以直接引用的网络很多,有全连接层、卷积层、循环层、归一化层、激活层等。
PyTorch构建模型大致有以下两种方式。
1)继承nn.Module基类来构建模型。
2)使用PyTorch Lightning来构建模型。