【记录1】常用的预训练pytorch模型下载地址

最新推荐文章于 2024-03-15 14:47:46 发布

原创最新推荐文章于 2024-03-15 14:47:46 发布 · 319 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#卷积神经网络 #pytorch #深度学习

记录专栏收录该内容

5 篇文章

订阅专栏

本文档汇总了PyTorch中常用的ImageNet预训练模型，包括ResNet系列（18、34、50、101、152）、DenseNet（121、161、169、201）和VGGNet（11、13、16、19及其bn版），供深度学习开发者快速上手迁移学习。

部署运行你感兴趣的模型镜像

ResNet系列在ImageNet上的预训练模型

    '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',

DenseNet

'densenet121': 'https://download.pytorch.org/models/densenet121-a639ec97.pth',
'densenet161': 'https://download.pytorch.org/models/densenet161-8d451a50.pth',
'densenet169': 'https://download.pytorch.org/models/densenet169-b2777c0a.pth',
'densenet201': 'https://download.pytorch.org/models/densenet201-c1103571.pth',

VGGNet

'vgg11': 'https://download.pytorch.org/models/vgg11-bbd30ac9.pth',
'vgg13': 'https://download.pytorch.org/models/vgg13-c768596a.pth',
'vgg16': 'https://download.pytorch.org/models/vgg16-397923af.pth',
'vgg19': 'https://download.pytorch.org/models/vgg19-dcbb9e9d.pth',
'vgg11_bn': 'https://download.pytorch.org/models/vgg11_bn-6002323d.pth',
'vgg13_bn': 'https://download.pytorch.org/models/vgg13_bn-abd245e5.pth',
'vgg16_bn': 'https://download.pytorch.org/models/vgg16_bn-6c64b313.pth',
'vgg19_bn': 'https://download.pytorch.org/models/vgg19_bn-c79401a0.pth',

其他

'inception_v3_google': 'https://download.pytorch.org/models/inception_v3_google-1a9a5a14.pth'

您可能感兴趣的与本文相关的镜像

PyTorch 2.6

PyTorch

Cuda

PyTorch 是一个开源的 Python 机器学习库，基于 Torch 库，底层由 C++ 实现，应用于人工智能领域，如计算机视觉和自然语言处理