aljaz/DALLE2-pytorch
1
1
Fork 0
mirror of https://github.com/lucidrains/DALLE2-pytorch.git synced 2026-02-23 20:25:00 +01:00
Code Issues Packages Projects Releases Wiki Activity

todo

Browse Source
This commit is contained in:
Phil Wang
2022-05-15 14:28:35 -07:00
committed by GitHub
parent 7b7a62044a
commit 0f0011caf0
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
README.md
View File

@@ -1013,6 +1013,7 @@ Once built, images will be saved to the same directory the command is invoked
- [ ] decoder needs one day worth of refactor for tech debt
- [ ] allow for unet to be able to condition non-cross attention style as well
- [ ] for all model classes with hyperparameters that changes the network architecture, make it requirement that they must expose a config property, and write a simple function that asserts that it restores the object correctly
- [ ] for both diffusion prior and decoder, all exponential moving averaged models needs to be saved and restored as well (as well as the step number)
## Citations