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