有关模型的保存、读取请参考:https://blog.csdn.net/weixin_42468475/article/details/110700546

有关模型的训练教程请参考:https://blog.csdn.net/weixin_42468475/article/details/110565705

torch.nn 里有很多神经网络模型需要的各种层和工具。

神经网络的工作流程:
在这里插入图片描述