训练模型时如何加载预训练权重

编程入门行业动态更新时间:2024-10-24 19:17:14

训练模型时如何加载预训练<a href=https://www.elefans.com/category/jswz/34/1758801.html style= 权重"/>

训练模型时如何加载预训练权重

这里主要介绍两种加载预训练权重的情况：
1）是所加载的预训练权重跟网络的初始权重是一一对应的，即字典的键值是对应的只是键的名称不同；
2）是所加载的预训练权重跟网络的初始权重不是一一对应的，但所加载的权重中有跟网络初始权重所对应的值

第一种情况

net_weights = net.dense.state_dict()   #读取网络的初始权重，以字典的形式存在net_weights中
pre_weights = torch.load(args.save_folder + args.basenet)  # 加载网络的网络的预训练权重，以字典的形式存在pre_weights中
for idx, key in enumerate(net_weights.keys(

更多推荐

训练模型时如何加载预训练权重

本文发布于:2024-02-08 20:39:50，感谢您对本站的认可！

本文链接:https://www.elefans.com/category/jswz/34/1674807.html