IDDM v1.1.0-stable
Note
In this version, The storage format of our model has completely changed, and the model needs to be retrained if the latest code is pulled.
If you don‘t like this version, you can download this tag (v1.0.2).
What's Changed
- Major update: model layer refactored; New network selection (unet and cspdarkunet); Updated the README by @chairc in #11
- Update: Encapsulate code to improve reusability. by @chairc in #12
- Update: Run GPU with custom settings. by @chairc in #13
- Add: Add show and save image in images. by @chairc in #14
- Update: Reconstruct model/networks and other included files. by @chairc in #15
- Update: Add function comments; Update model.parameters() and add SGD optimizer. by @chairc in #16
- Major update: Reconstruct modules.py, and update package path; Rewrite checkpoint storage and loading functions, add checkpoint files by @chairc in #18
New Contributors
Full Changelog: v1.0.2...v1.1.0
About weight model
NEU weight parameters is {"conditional"=True, "image_size"=64, "sample"=ddim, "network"=unet, "act"=gelu, "num_classes"=6}
Cifar10 weight parameters is {"conditional"=True, "image_size"=64, "sample"=ddim, "network"=unet, "act"=gelu, "num_classes"=10}
This weight include model , ema model and optimizer checkpoints.