ViT를 pytorch 환경에서 구현한 사이트

https://github.com/lucidrains/vit-pytorch

위 사이트를 이용하면 Vision Transformer(ViT)를 pytorch에서 구현한 코드들을 손쉽게 활용가능하다.

설치는 간단하게 pip install vit-pytorch로 가능하며

예제들을 이용한 활용 방법 또한 설명되어 있다.