文章原创自:微信公众号「机器学习炼丹术」 作者:炼丹兄 联系方式:微信cyx645016617 代码来自github [前言]:看代码的时候,也许会不理解VIT中各种组件的含义,但是这个文章的目的是了解其实现.在之后看论文的时候,可以做到心中有数,而不是一片茫然. VIT类 初始化 和之前的学习一样,从大模型类开始看起,然后一点一点看小模型类: class ViT(nn.Module): def __init__(self, *, image_size, patch_size, num_clas…