wulele2的博客

前言断断续续写了好久，本篇是最后一篇。太肝了，写这东西。文章目录前言一、创建一个简单网络并训练保存二、State_dict()源码解读三、加载模型四、实战：保存/加载（模型+优化器+epoch）五、网络相同但命名方式不同的加载解析六、网络不同的权重加载解析附录、其他常用的代码片段参考文献一、创建一个简单网络并训练保存import torchimport torch.nn as nnimport torch.nn.functional as Ffrom torch.autograd i

Module源码介绍本篇是nn.Module源码系列文章介绍第二篇，建议读者在阅读第一篇之后，在来阅读本篇。当然，也可以直接阅读本文，因为提供了大量的实例。第一篇地址文章目录Module源码介绍前置知识：Module中train/eval模块状态切换实战：随意进行train/eval状态切换实战：冻结网络中所有BN层nn.Module中指定梯度和梯度清0函数实战：冻结BN层梯度参数总结前置知识：Module中train/eval模块状态切换  在上篇文章中，介绍了nn.Module是如何完成

目录文章目录目录前言一、构造一个极简网络二、使用步骤1.引入库2.读入数据总结前言一、构造一个极简网络import torchimport torch.nn as nnfrom collections.abc import Iterable, Iteratorclass Net(nn.Module): def __init__(self): super(Net, self).__init__() def forward(self,x):