1. 首页
  2. 编程语言
  3. Python
  4. pytorch 在网络中添加可训练参数,修改预训练权重文件的方法

pytorch 在网络中添加可训练参数,修改预训练权重文件的方法

上传者: 2022-03-31 01:16:50上传 PDF文件 64.27 KB 热度 13次

实践中,针对不同的任务需求,我们经常会在现成的网络结构上做一定的修改来实现特定的目的。如果我们有原先网络的预训练权重,现在添加了一个新的参数,原有的权重文件自然就不能加载了,我们需要修改原权重文件,在其中添加我们的新变量的初始值。现在权重就可以加载在修改后的模型上了。

用户评论