diff --git a/README.md b/README.md index 6c6e8b6..bf6e3ae 100644 --- a/README.md +++ b/README.md @@ -827,6 +827,7 @@ Once built, images will be saved to the same directory the command is invoked - [ ] train on a toy task, offer in colab - [ ] 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 ## Citations