document in readme later

2025-12-30 08:02:29 +00:00 · 2023-10-03 09:26:02 -07:00
parent 3fdb8dd352
commit 680d446e46
1 changed files with 5 additions and 0 deletions
--- a/vit_pytorch/simple_vit_with_register_tokens.py
+++ b/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