预训练权重加载

一般为了缩短网络训练时间,并且希望达到更好的精度,我们一般加载预训练权重进行网络的训练。
加载的过程中可能会遇到,网络权重与网络层命名不同的情况。

目录

  • 一、 网络结构与权重完全对应
  • 二、网络结构相同,网络层命名不同

一、 网络结构与权重完全对应


model.load_state_dict(torch.load(weight_pth))

二、网络结构相同,网络层命名不同

def load_weight(net, weight):
	weight_file = '/weight_pth'
	dic={}
	for now_keys, values in zip(net.keys(), torch.load(weight_file).values())
		dic[now_kyes]=values
	net.load_state_dict(dic)
	return net

你可能感兴趣的:(目标检测,实时检测,深度学习)