.. py:module::  diffusion.model.diffusion
.. currentmodule:: diffusion.model.diffusion

===============
model.diffusion
===============

.. autoclass:: diffusion.model.DenoiseDiffusion
   :members: q_sample, p_sample, loss