torchvision.models as models>>> model = models.vit_bFile "<stdin>", line 1, in <module>
AttributeError: module 'torchvision.models' has no attribute 'vit_b
我是新的拥抱脸,并希望采用相同的变压器架构做在ViT的图像分类到我的领域。因此,我需要改变输入形状和所做的增强。requests.get(url, stream=True).raw)
model = TFViTForImageClassification.from_pretrained("google/vit-base