intent

lucidrains · lucidrains · commit 4c5726f7c227 · 2021-12-03T12:18:12.000-08:00
diff --git a/README.md b/README.md
@@ -152,6 +152,8 @@ quantized, indices, loss = vq(img_fmap) # (1, 256, 32, 32), (1, 32, 32), (1,)
 ## Todo
 
 - [ ] allow for multi-headed codebooks, from https://openreview.net/forum?id=GxjCYmQAody
+- [ ] support masking
+
 
 ## Citations
 
