有关模型的保存、读取请参考:https://blog.csdn.net/weixin_42468475/article/details/110700546
有关模型的训练教程请参考:https://blog.csdn.net/weixin_42468475/article/details/110565705
torch.nn 里有很多神经网络模型需要的各种层和工具。
神经网络的工作流程:
直接开始正题
conv2d 卷积层参数的解释:
conv2 = nn.Conv2d(in_channels=