使用pytorch从头实现一个vit
我们实现的目标是做image classification,使用MINIST数据集。 首先把必要的库导入进来。 import numpy as npfrom tqdm import tqdm, trangeimport torch import torch.nn as nn from torch.optim import Adam from torch.nn import CrossEntropyLoss from torch.…
2026-05-23