IDDM v1.1.2-stable
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