Skip to content

IDDM v1.0.1
Compare
Choose a tag to compare
@chairc chairc released this 14 Aug 17:48
· 254 commits to main since this release
v1.0.1
3089bb8

We are delighted to announce that version 1.0.1 of IDDM (Industrial Defect Diffusion Model) has been officially released.
In this version, we have made the following changes.

What's Changed

  • Fix the bug that the model does not save when a single card or CPU is used. #5ee6010
  • Fix resuming training model and optimizer parameters loading. #e58cd4e
  • Modify the resume training model loading and cancel the broadcast method. #19bc4d9
  • Add seed initializer. #9189913
  • Add learning rate function such as default, cosine and warmup_cosine. #35d024e
  • Add network summary test function. #58f0694
  • Modify some formatting issues. #f35040b
  • Add requirements.txt #0fb164f
  • Refactoring the sampling model structure by creating a base class for common parts and having other sampling methods inherit from it.#278debd
  • Add DDIM sample method. #c469e4f
  • Add sample selector method during training. #6acf6cc
  • Update README #329ded9
  • Add image results generated in README. #5df2947
  • Modify Repository Structure. #6c21b9b
  • Optimizing loading weight method, optimizing generate.py and Check for issues where distributed weight names are not the same as network weights. #24a03c3
  • Edited to english in 'train.py' and 'test_module.py'. #788138d
  • Edited to english in 'train.py' and 'generate.py'. #f2e9d7c
  • Edited to english in 'network.py' and 'modules.py'. #8c0bb55
  • Edited to english in 'base.py', 'ddpm.py' and 'ddim.py'. #c92fd82
  • Edit english in 'generate.py'. #970ec3c
  • Edit untranslated parts in 'utils.py'. #3089bb8

Full Changelog: v1.0.0...v1.0.1

Assets 2
Loading