<svg xmlns="http://www.w3.org/2000/svg" style="display: none;"> <path stroke-linecap="round" d="M5,0 0,2.5 5,5z" id="raphael-marker-block" style="-webkit-tap-highlight-color: rgba(0, 0, 0, 0);"></path> </svg> <h4>一、模型的保存与读取</h4> 讯享网
1.方法一：保存整个模型结构（无需加载py文件）
讯享网
2.方法二：只保存权重（py文件包括一个网络结构）
讯享网
3.保存权重，优化器等参数
二、对模型进行修改
适用于：只读取pth中的部分权重（不需要分类器层），或是没有网络模型结构，需要对模型进行调整时（如剪枝情况下，不能照搬剪枝前模型的网络结构）
1.加载部分权重
讯享网
2.继承老模型并修改
方法一：修改模型的前向传播逻辑
加载原始模型后，直接修改模型的前向传播方法，使其在前向传播过程中不经过要删除的层。
方法二：创建新模型并复制参数（除特定层外）
定义一个新的模型类，它继承自原始模型类，但在构造函数中不创建要删除的层。
加载原始模型的状态字典，然后遍历状态字典，将除了要删除的层之外的参数复制到新模型中。
例如
讯享网
三、使用模型进行预测
1.创建一个张量
2.对图像进行预测
讯享网
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容，请联系我们，一经查实，本站将立刻删除。
如需转载请保留出处：https://51itzy.com/kjqy/173343.html