From c87b84a25908e846c520746a8e7baa794a7a46db Mon Sep 17 00:00:00 2001 From: Phil Wang Date: Sat, 7 May 2022 09:21:08 -0700 Subject: [PATCH] todo --- README.md | 1 + 1 file changed, 1 insertion(+) diff --git a/README.md b/README.md index 9f1e8e9..7b4c0a8 100644 --- a/README.md +++ b/README.md @@ -982,6 +982,7 @@ Once built, images will be saved to the same directory the command is invoked - [ ] make sure resnet hyperparameters can be configurable across unet depth (groups and expansion factor) - [ ] offer save / load methods on the trainer classes to automatically take care of state dicts for scalers / optimizers / saving versions and checking for breaking changes - [ ] offer setting in diffusion prior to split time and image embeddings into multiple tokens, configurable, for more surface area during attention +- [ ] bring in skip-layer excitatons (from lightweight gan paper) to see if it helps for either decoder of unet or vqgan-vae training ## Citations