document in readme later

Author: lucidrains

vit_pytorch/simple_vit_with_register_tokens.py

@@ -1,3 +1,8 @@
+"""
+    Vision Transformers Need Registers
+    https://arxiv.org/abs/2309.16588
+"""
+
 import torch
 from torch import nn