借助Kaggle快速下载PyTorch预训练模型

最新推荐文章于 2024-05-07 21:21:54 发布

「已注销」

最新推荐文章于 2024-05-07 21:21:54 发布

阅读量716

点赞数 1

文章标签： pytorch kaggle

本文链接：https://blog.csdn.net/HUNLUAN3004/article/details/104747468

版权

本文介绍了如何使用Kaggle来快速下载PyTorch的预训练模型，避免默认方式下的慢速下载问题。通过链接提供了一个Kaggle notebook，该notebook在Kaggle的output中生成预训练模型的下载链接。下载模型后，将其放置于指定位置即可供使用。

摘要由CSDN通过智能技术生成

pytorch里的预训练模型，使用默认的方式下载会非常慢。最近尝试了借助Kaggle下载。
这里有所有模型的列表 https://github.com/pytorch/vision/tree/master/torchvision/models

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'
    'inception_v3_google': 'https://download.pytorch.org/models/inception_v3_google-1a9a5a14.pth',
    'densenet121': 'https://download.pytorch.org/models/densenet121-a639ec97.pth',
    'densenet169': 'https://download.pytorch.org/models/densenet169-b2777c0a.pth',
    'densenet201': 'https://downlo