diff --git a/README.md b/README.md index bf6e3ae..539f55b 100644 --- a/README.md +++ b/README.md @@ -828,6 +828,7 @@ Once built, images will be saved to the same directory the command is invoked - [ ] think about how best to design a declarative training config that handles preencoding for prior and training of multiple networks in decoder - [ ] extend diffusion head to use diffusion-gan (potentially using lightweight-gan) to speed up inference - [ ] bring in cross-scale embedding from iclr paper https://github.com/lucidrains/vit-pytorch/blob/main/vit_pytorch/crossformer.py#L14 +- [ ] figure out if possible to augment with external memory, as described in https://arxiv.org/abs/2204.11824 ## Citations