Skip to content

IDDM v1.1.0-stable

Compare
Choose a tag to compare
@chairc chairc released this 05 Dec 16:23
· 199 commits to main since this release
d94531b

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.