pytorch 26 科学炼丹（训练）方式之——魔改网络（给模型做外科手术，实现module的删除、修改与添加，forward流程的修改）_如何魔改网络-CSDN博客

网友收藏2023-10-13 01:47

链接地址：https://blog.csdn.net/a486259/article/details/124805102
链接标题：pytorch 26 科学炼丹（训练）方式之——魔改网络（给模型做外科手术，实现module的删除、修改与添加，forward流程的修改）_如何魔改网络-CSDN博客
所属网站：blog.csdn.net
被收藏次数：3675

网络结构层出不穷，但经典网络总是大家工程上的最爱。有的时候需要对网络进行微操，直接该源码一是工作量太大；二是破坏了原来的网络结构会导致预训练权重无法加载。因此，可以在加载预训练权重后对模型进行微操。本博文实现了模型中module的删除、修改与添加、以及模型forward流程的修改（重定义），后续将会以模型库中预定义的AlexNet为例子进行实操。1、module的添加该操作通过model.add_module实现，具体可见下列代码示意import torchfrom torch impor_如何魔改网络

本文地址：https://tebull.com/detail/85402.html

标签：如何魔改网络