Skip to content

v1.2

Choose a tag to compare

@kisonho kisonho released this 25 Jul 18:09
· 6 commits to main since this release

API Updates:

  • Add EMA support in Manager
  • Add multiple devices support in training configs
  • Display improvement during testing
  • FP16 support
  • Gradients accumulative support
  • Introducing callbacks.SamplingCallback to sample images during training
  • Introducing optim.EMAOptimizer for EMA optimizer with context control to swap parameters

Other Updates:

  • Deprecated python 3.9
  • Typing improvement