使用pytorch进行迁移学习模型下载失败解决办法

最新推荐文章于 2024-07-15 20:55:12 发布

haohulala

最新推荐文章于 2024-07-15 20:55:12 发布

阅读量6.1k

点赞数 14

分类专栏：计算机视觉

本文链接：https://blog.csdn.net/haohulala/article/details/107598859

版权

使用pytorch进行迁移学习的时候，我们需要下载预训练的模型，但是这个模型通常很大，如果在代码中在线下载的话，很可能会中断，并且一中断之前也就白下载了，这篇文章里我介绍一种离线使用预训练模型的方法。

所谓离线使用预训练模型的方法，实际上就是使用浏览器将模型下载下来（通常浏览器下载会比较稳定，并且如果下载中断还能恢复），下面给出各种模型的下载地址，只需要将对应的url键入到浏览器中就可以建立下载

1. Resnet:
  model_urls = {
      'resnet18': 'https://download.pytorch.org/models/resnet18-5c106cde.pth',
      'resnet34': 'https://download.pytorch.org/models/resnet34-333f7ec4.pth',
      'resnet50': 'https://download.pytorch.org/models/resnet50-19c8e357.pth',
      'resnet101': 'https://download.pytorch.org/models/resnet101-5d3b4d8f.pth',
      'resnet152': 'https://download.pytorch.org/models/resnet152-b121ed2d.pth',
  }

2. inception:
 model_urls = {
      Inception v3 ported from TensorFlow
     'inception_v3_google': 'https://download.pytorch.org/models/inception_v3_google-