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

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

第一种情况

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