Contents Menu Expand Light mode Dark mode Auto light/dark mode
dmme 0.5.2 documentation
dmme 0.5.2 documentation

PAPERS:

  • DDPM
  • DDIM

API REFERENCE:

  • dmme
  • dmme.lit_modules
    • LitDDPM
    • LitDDIM
    • LitIDDPM
  • dmme.diffusion_models
    • DDPM
    • DDIM
    • IDDPM
  • dmme.equations
    • dmme.equations.ddpm
    • dmme.equations.ddim
    • dmme.equations.iddpm
  • dmme.models
    • U-Net from Denoising Diffusion Probabilistic Models
    • Improved U-Net from Improved Denoising Diffusion Models
  • dmme.datasets
    • LSUN
  • dmme.data_modules
    • DataModule
    • CIFAR10
    • LSUN
  • dmme.callbacks
    • EMA
    • GenerateImage
  • dmme.trainer
Back to top
Edit this page

dmme.trainer#

dmme.trainer.main()[source]#

CLI entrypoint

refer to LightningCLI docs for more information

Previous
GenerateImage
Copyright © 2022, Chanhyuk Jung
Made with Sphinx and @pradyunsg's Furo
On this page
  • dmme.trainer
    • main()