pytorch 加载本地模型

最新推荐文章于 2024-10-08 17:14:50 发布

原创

最新推荐文章于 2024-10-08 17:14:50 发布 · 2k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#pytorch #deep learning

本文介绍了如何在PyTorch中加载本地的VGG11模型。首先，可以从PyTorch官网获取模型下载链接，然后参照教程进行本地加载。详细步骤包括模型的下载地址和本地加载的方法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

模型下载

从pytorch官网找到对应模型的下载地址，并使用浏览器下载 https://pytorch.org/hub/

模型加载(VGG11为例)

vgg11下载地址
https://github.com/pytorch/vision/blob/master/torchvision/models/vgg.py

本地加载

import torch
import torchvision.models as models
model = models.vgg11(pretrained=False)
model.load_state_dict(torch.load(R'pytorch\vgg11-8a719046.pth'))
print(model.eval())

# 输出
VGG(
  (features): Sequential(
    (0): Conv2d(3, 64, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1))
    (1): ReLU(inplace=True)
    (2): MaxPool2d(kernel_size=2, stride=2, padding=0, dilation=1, ceil_mode=False)
    (3): Conv2d(64, 128, kerne