解决torchvision.model下载预训练模型太慢的问题

以下代码等价,但第一种慢,第二种快:

import torchvision.models as models

self.resnet = models.resnet18(pretrained=True)  #  联网下载,慢
self.model = models.resnet18(pretrained=False)
state_dict = torch.load('src/model/resnet18-5c106cde.pth')  # 自己从网上下载.pth,快
self.model.load_state_dict(state_dict)  # 再把读出来的参数放进没有参数的模型

当pretrained=True,才会联网下载模型,否则很快,仅得到一个没训练过的模型。

.pth文件或者state_dict变量:模型参数,里面是模型每一层具体的浮点数

model:模型,不含参数

model和.pth如果是对应的,就可以用model.load_state_dict加载。注意这条语句是在模型上直接修改,不应写成model = model.load_state_dict。

所以我们可以自己在浏览器下载模型,然后加载进去。那么去哪里下载呢?Ctrl+函数打开源码自己就可以找到。

相关推荐

  1. 解决torchvision.model下载训练模型问题

    2024-01-16 17:20:07       55 阅读
  2. 训练模型下载和使用

    2024-01-16 17:20:07       67 阅读
  3. 解决 Windows 下 Git 下载问题

    2024-01-16 17:20:07       47 阅读

最近更新

  1. docker php8.1+nginx base 镜像 dockerfile 配置

    2024-01-16 17:20:07       94 阅读
  2. Could not load dynamic library ‘cudart64_100.dll‘

    2024-01-16 17:20:07       100 阅读
  3. 在Django里面运行非项目文件

    2024-01-16 17:20:07       82 阅读
  4. Python语言-面向对象

    2024-01-16 17:20:07       91 阅读

热门阅读

  1. 【git】如何提交pull request

    2024-01-16 17:20:07       43 阅读
  2. http 请求流程,网络的七层模型

    2024-01-16 17:20:07       51 阅读
  3. 在QT中使用POST方式发送HTTP数据

    2024-01-16 17:20:07       56 阅读
  4. 如何为 glog 的宏重载 <<

    2024-01-16 17:20:07       56 阅读