如何使用PyTorch从本地加载.pth模型文件？
- 内容介绍
- 文章标签
- 相关推荐
本文共计376个文字，预计阅读时间需要2分钟。
由于下载预训练模型可能因网络速度慢而引发连接错误，以下是一个简化的代码示例，使用本地模型替代预训练模型：
pythonimport torchvision.models as models
使用本地预训练模型model=models.vgg16(pretrained=False)print(model)
可以从官网加载预训练好的模型：
import torchvision.models as models model = models.vgg16(pretrained = True) print(model)
但是经常会出现因为下载速度太慢而出现requests.exceptions.ConnectionError: ('Connection aborted.', TimeoutError(10060, '由于连接方在一段时间后没有正确答复或连接的主机没有反应，连接尝试失败。', None, 10060, None))这种错误，因此需要我们手动去下载 .pth 文件（百度云也很慢，如果你是SVIP，当我没说；迅雷的速度也还可以），然后从本地加载。
本文共计376个文字，预计阅读时间需要2分钟。
由于下载预训练模型可能因网络速度慢而引发连接错误，以下是一个简化的代码示例，使用本地模型替代预训练模型：
pythonimport torchvision.models as models
使用本地预训练模型model=models.vgg16(pretrained=False)print(model)
可以从官网加载预训练好的模型：
import torchvision.models as models model = models.vgg16(pretrained = True) print(model)
但是经常会出现因为下载速度太慢而出现requests.exceptions.ConnectionError: ('Connection aborted.', TimeoutError(10060, '由于连接方在一段时间后没有正确答复或连接的主机没有反应，连接尝试失败。', None, 10060, None))这种错误，因此需要我们手动去下载 .pth 文件（百度云也很慢，如果你是SVIP，当我没说；迅雷的速度也还可以），然后从本地加载。