add diagram

lucidrains · Mar 5, 2021 · 0705deb · 0705deb
1 parent 2e2a9cd
commit 0705deb
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -1,4 +1,6 @@
-## Perceiver - Pytorch (wip)
+<img src="./perceiver.png" width="600px"></img>
+
+## Perceiver - Pytorch
 
 Implementation of <a href="https://arxiv.org/abs/2103.03206">Perceiver</a>, General Perception with Iterative Attention, in Pytorch
 
@@ -31,7 +33,7 @@ model = Perceiver(
 
 img = torch.randn(1, 224 * 224) # 1 imagenet image, pixelized
 
-model(img)
+model(img) # (1, 1000)
 ```
 ## Citations
 

diff --git a/perceiver.png b/perceiver.png