diff --git a/README.md b/README.md index 6f92100..ff2b833 100644 --- a/README.md +++ b/README.md @@ -1,4 +1,4 @@ -# VisionTransformer +# VisionTransformer (Pytorch) A complete easy to follow implementation of Google's Vision Transformer proposed in "AN IMAGE IS WORTH 16X16 WORDS". This pytorch implementation has comments for better understanding. **An image is worth 16x16 words: transformers for image recognition at scale:**