pytorch实现从本地加载 .pth 格式模型
可以从官网加载预训练好的模型：
importtorchvision.modelsasmodels model=models.vgg16(pretrained=True) print(model)
但是经常会出现因为下载速度太慢而出现requests.exceptions.ConnectionError:('Connectionaborted.',TimeoutError(10060,'由于连接方在一段时间后没有正确答复或连接的主机没有反应，连接尝试失败。',None,10060,None))这种错误，因此需要我们手动去下载.pth文件（百度云也很慢，如果你是SVIP，当我没说；迅雷的速度也还可以），然后从本地加载。
从本地加载只需要把上面的代码换成如下：
importtorchvision.modelsasmodels model=models.vgg16(pretrained=False) pre=torch.load(r'.\kaggle_dog_vs_cat\pretrain\vgg16-397923af.pth') model.load_state_dict(pre)
如果你模型不是用的vgg16，而是用的vgg11或者vgg13，只需要修改语句model=models.vgg16(pretrained=False)为对应模型的函数即可。
以上这篇pytorch实现从本地加载.pth格式模型就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持毛票票。
声明：本文内容来源于网络，版权归原作者所有，内容由互联网用户自发贡献自行上传，本网站不拥有所有权，未作人工编辑处理，也不承担相关法律责任。如果您发现有涉嫌版权的内容，欢迎发送邮件至：czq8825#qq.com（发邮件时，请将#更换为@）进行举报，并提供相关证据，一经查实，本站将立刻删除涉嫌侵权内容。