From 0b4ec34efb2efe140e574254288296110f1c6002 Mon Sep 17 00:00:00 2001 From: Phil Wang Date: Wed, 20 Apr 2022 12:24:23 -0700 Subject: [PATCH] todo --- README.md | 1 + 1 file changed, 1 insertion(+) diff --git a/README.md b/README.md index 063caab..3bb1940 100644 --- a/README.md +++ b/README.md @@ -415,6 +415,7 @@ Offer training wrappers - [ ] build out latent diffusion architecture, make it completely optional (additional autoencoder + some regularizations [kl and vq regs]) (figure out if latent diffusion + cascading ddpm can be used in conjunction) - [ ] Extend diffusion head to use diffusion-gan (potentially using lightweight-gan) to speed up inference - [ ] become an expert with unets, cleanup unet code, make it fully configurable, port all learnings over to https://github.com/lucidrains/x-unet +- [ ] copy the cascading ddpm code to a separate repo (perhaps https://github.com/lucidrains/denoising-diffusion-pytorch) as the main contribution of dalle2 really is just the prior network - [ ] train on a toy task, offer in colab ## Citations