vision-understanding ·

vit-pytorch

Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch

原始来源：https://github.com/lucidrains/vit-pytorch