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