如何在 pytorch 中加载预训练的 googlenet 模型

我正在尝试通过特定数据集对 GoogleNet 网络进行微调,但在加载时遇到问题。我现在尝试的是:

model = torchvision.models.googlenet(pretrained=True)

但是我收到一个错误:

AttributeError: module 'torchvision.models' has no attribute 'googlenet'

我有最新版本的torchvision,但为了确定重新安装,错误仍然存​​在。

kmario23

您可以改为使用 GoogLeNetinception_v3模型(“重新思考计算机视觉的初始架构”):

import torchvision
google_net = torchvision.models.inception_v3(pretrained=True)

本文收集自互联网,转载请注明来源。

如有侵权,请联系 [email protected] 删除。

编辑于
0

我来说两句

0 条评论
登录 后参与评论

相关文章

如何在gensim中加载预训练模型并对其进行训练doc2vec?

如何在 pytorch 中的修改后的 vgg19 网络中加载预训练权重?

如何在具有 .npy 扩展名的 gensim 中加载预训练的 fastText 模型

如何从Pytorch中的预训练模型加载保存的令牌生成器

使用bvlc_googlenet作为数字的预训练模型-错误

如何在 keras 中使用预训练的 googlenet 和 alexnet

使用Pytorch从.ckpt文件加载预训练(CNN)模型

如何更改pytorch预训练模型中的类数

如何在 Pytorch 中的 EfficieNet 预训练模型中添加最后一个分类层?

如何在 Django 中加载经过训练的模型

如何在pytorch中加載自定義模型

PyTorch / Gensim-如何加载预训练的单词嵌入

如何在 Colab 的 PyTorch 中加载在函数中定义的已保存模型?

如何使用PyTorch在预训练模型上添加新图层?(给出了Keras示例。)

在PyTorch中加载受Torch7训练的模型(.t7)

在聊天机器人的 pytorch 中加载训练好的模型保存

GoogLeNet模型的微调

如何在COCO日期集上加载经过预训练的模型以进行图像分割?

加载预训练的 fasttext 模型

微调预训练模型 MobileNet_V3_Large PyTorch

是否可以从GCS存储桶URL加载预训练的Pytorch模型而无需先在本地持久化?

如何在Pytorch预训练模块中更改激活层?

如何从训练 Pytorch 和 PytorchText 模型进行预测?

如何使用PyTorch训练原始的U-Net模型?

如何在Pytorch中检查模型是否处于训练或评估模式?

如何在A框架中加载.glb模型?

如何在Laravel中加载模型?

如何在TDB TripleStore中加载模型

如何在 PyTorch 中保存模型?