Shortcuts

mmedit.models.editors.ddpm.ddpm_scheduler

Module Contents

Classes

DDPMScheduler

class mmedit.models.editors.ddpm.ddpm_scheduler.DDPMScheduler(num_train_timesteps: int = 1000, beta_start: float = 0.0001, beta_end: float = 0.02, beta_schedule: str = 'linear', trained_betas: Optional[Union[numpy.array, list]] = None, variance_type='fixed_small', clip_sample=True)[源代码]
set_timesteps(num_inference_steps)[源代码]

set timesteps.

_get_variance(t, predicted_variance=None, variance_type=None)[源代码]

get variance.

step(model_output: torch.FloatTensor, timestep: int, sample: torch.FloatTensor, predict_epsilon=True, cond_fn=None, cond_kwargs={}, generator=None)[源代码]
add_noise(original_samples, noise, timesteps)[源代码]

add noise.

abstract training_loss(model, x_0, t)[源代码]
abstract sample_timestep()[源代码]
__len__()[源代码]
Read the Docs v: latest
Versions
master
latest
stable
zyh-doc-notfound-extend
Downloads
pdf
epub
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.