首先,我们需要安装一个库。 pip install vit-pytorch -i https://pypi.tuna.tsinghua.edu.cn/simple pip install timm -i https://pypi.tuna.tsinghua.edu.cn/simple 然后就可以在代码中使用Vit了 fromvit_pytorchimportViTimporttorch net=ViT(image_size=224,patch_size=32,num_classes=1000,dim=1024,depth...