Update KNNMemory.add to correctly index and update KNN list, after ad…

lucidrains / memorizing-transformers-pytorch

Implementation of Memorizing Transformers (ICLR 2022), attention net augmented with indexing and retrieval of memories using approximate nearest neighbors, in Pytorch

MIT License

620 stars 46 forks source link

Update KNNMemory.add to correctly index and update KNN list, after ad… #11

Closed LakshyAAAgrawal closed 1 year ago

LakshyAAAgrawal commented 1 year ago

…ding new memories in parallel

lucidrains commented 1 year ago

ohh thanks!