Skip to content

IDDM v1.1.2-stable

Compare
Choose a tag to compare
@chairc chairc released this 12 Jan 09:48
· 165 commits to main since this release
bf24b96

What's Changed

  • Update: Modify automatic mixed precision training name; reconstruct automatic mixed precision training structure. by @chairc in #28
  • Update: Modify project's name. by @chairc in #29
  • Add: Add current epoch average loss log. by @chairc in #30
  • Update: Update the introduction in README. by @chairc in #31
  • Major update about generating images by @chairc in #33
  • Modify some parameters and formats; Generate images in new folder. by @chairc in #34
  • Modify comment; Modify README, add issue link and correct some text. by @chairc in #35

Note

Due to changes in the model saving structure, the following parameters in 'generate.py' may not be entered after this version.

Parameter Name Conditional Usage Type Description
--sample Sampling method str Set the sampling method type, currently supporting DDPM and DDIM. (No need to set for models after version 1.1.1)
--network Training network str Set the training network, currently supporting UNet, CSPDarkUNet. (No need to set for models after version 1.1.1)
--act Activation function str Activation function selection. Currently supports gelu, silu, relu, relu6 and lrelu. If you do not set the same activation function as the model, mosaic phenomenon will occur. (No need to set for models after version 1.1.1)
--num_classes Number of classes int Number of classes for classification (No need to set for models after version 1.1.1)

Full Changelog: v1.1.1...v1.1.2