预训练权重加载
一般为了缩短网络训练时间,并且希望达到更好的精度,我们一般加载预训练权重进行网络的训练。加载的过程中可能会遇到,网络权重与网络层命名不同的情况。目录一、 网络结构与权重完全对应二、网络结构相同,网络层命名不同一、 网络结构与权重完全对应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 ·
3707 阅读 ·
5 评论