AttributeError: type object ‘VisionTransformer‘ has no attribute ‘from_pretrained‘

错误原因

VisionTransformer本身并没有from_pretrained方法。但是,如果您希望使用预训练的VisionTransformer模型,可以使用timm库中的create_model函数来加载模型。

解决办法:

import torch
import timm

model = timm.create_model('vit_base_patch16_224', pretrained=True)

猜你喜欢

转载自blog.csdn.net/weixin_44575717/article/details/129092323