Pytorch加载预先训练模型时下载出现错误

Pytorch加载预先训练模型时下载出现错误

加载预先训练模型出现问题

背景：我在做Transfer style的pytorch官网demo的时候，想使用预先训练好的VGG19模型参数：

cnn = dels.vgg19(pretrained=True)

出现问题是下载特别特别慢，几个小时只有10%左右。在这里记录一下我最终的解决方案。

首先参照

这篇文章博主的解决方案。果然，成功了一半。但是在加载的过程中还是出现了数据格式不匹配的问题。这时候，在原方案的基础上，应进行如下更改：

cnn = models.vgg19(pretrained=False).(device).eval()
pthfile = r’C:UserschengDesktop【02】ENCODER_DECODERvgg19-dcbb9e9d.pth’
pre = torch.load(pthfile) # Pre-trained Parameters
cnn.load_state_dict(torch.load(pthfile),strict=False)

也就是说在加载模型的时候，使用strcit=False, 这样数据不需要严格一致。问题解决。