1. 首页
  2. 数据库
  3. 其它
  4. pytorch实现从本地加载 .pth 格式模型

pytorch实现从本地加载 .pth 格式模型

上传者: 2020-12-22 23:07:06上传 PDF文件 38.55KB 热度 12次
可以从官网加载预训练好的模型: import torchvision.models as models model = models.vgg16(pretrained = True) print(model) 但是经常会出现因为下载速度太慢而出现requests.exceptions.ConnectionError: (‘Connection aborted.’, TimeoutError(10060, ‘由于连接方在一段时间后没有正确答复或连接的主机没有反应,连接尝试失败。’, None, 10060, None))这种错误,因此需要我们手动去下载 .pth 文件(百度云也很慢,如果你是S
用户评论