加载模型和权重

千寻渊渊

于 2021-03-27 15:41:52 发布

阅读量492

点赞数

文章标签：深度学习 python 神经网络 pytorch

本文链接：https://blog.youkuaiyun.com/m0_37613352/article/details/115265996

版权

model1 = VGG('vgg16')# 实例化自己的模型；
model1 = torch.nn.DataParallel(model1)
model1_weights = torch.load('vgg16_eig_pretrained.pt')
model1.load_state_dict({k.replace(".basic_model", ""):v for k, v in torch.load('vgg16_eig_pretrained.pt').items()})