出错代码
解决方法:下载需要加载的模型,然后修改为
resnet = models.resnet34(weights=None)
resnet.load_state_dict(torch.load('D:/Deepl/networks/resnet34-333f7ec4.pth'))
出错代码
解决方法:下载需要加载的模型,然后修改为
resnet = models.resnet34(weights=None)
resnet.load_state_dict(torch.load('D:/Deepl/networks/resnet34-333f7ec4.pth'))