integrations.diffusion.callbacks
integrations.diffusion.callbacks
Callbacks for diffusion training.
Classes
| Name | Description |
|---|---|
| DiffusionGenerationCallback | Callback for generating samples during diffusion training. |
DiffusionGenerationCallback
integrations.diffusion.callbacks.DiffusionGenerationCallback(trainer)Callback for generating samples during diffusion training.
Methods
| Name | Description |
|---|---|
| on_step_end | Generate samples at specified intervals. |
on_step_end
integrations.diffusion.callbacks.DiffusionGenerationCallback.on_step_end(
args,
state,
control,
**kwargs,
)Generate samples at specified intervals.