预训练权重加载
一般为了缩短网络训练时间,并且希望达到更好的精度,我们一般加载预训练权重进行网络的训练。
加载的过程中可能会遇到,网络权重与网络层命名不同的情况。
目录一、 网络结构与权重完全对应二、网络结构相同,网络层命名不同
一、 网络结构与权重完全对应
model.load_state_dict(torch.load(weight_pth))
二、网络结构相同,网络层命名不同
def load_weight(net, weight):
weight_file = '/weight_pth'
dic={}
f
原创
2021-03-03 21:32:40 ·
3715 阅读 ·
5 评论