错误原因:
VisionTransformer本身并没有from_pretrained方法。但是,如果您希望使用预训练的VisionTransformer模型,可以使用timm库中的create_model函数来加载模型。
解决办法:
import torch
import timm
model = timm.create_model('vit_base_patch16_224', pretrained=True)
错误原因:
VisionTransformer本身并没有from_pretrained方法。但是,如果您希望使用预训练的VisionTransformer模型,可以使用timm库中的create_model函数来加载模型。
解决办法:
import torch
import timm
model = timm.create_model('vit_base_patch16_224', pretrained=True)