我正在尝试通过特定数据集对 GoogleNet 网络进行微调,但在加载时遇到问题。我现在尝试的是:
model = torchvision.models.googlenet(pretrained=True)
但是我收到一个错误:
AttributeError: module 'torchvision.models' has no attribute 'googlenet'
我有最新版本的torchvision,但为了确定重新安装,错误仍然存在。
您可以改为使用 GoogLeNetinception_v3
模型(“重新思考计算机视觉的初始架构”):
import torchvision
google_net = torchvision.models.inception_v3(pretrained=True)
本文收集自互联网,转载请注明来源。
如有侵权,请联系 [email protected] 删除。
我来说两句