包括与PyTorch相关的库(torch),与数据处理相关的库(如torchvision)、与张量操作方面的库(如einops)等。2)对数据进行预处理。使用torchvision导入数据集CIFAR10,然后对...
浏览 52 次 标签: python之使用ViT进行图像分类