Index

A

alpha_bars (BaseBetaScheduler attribute) (BetaSchedulerConfiguration attribute)

B

BaseBetaScheduler (class in diffusion_models.gaussian_diffusion.beta_schedulers) BaseDiffuser (class in diffusion_models.gaussian_diffusion.base_diffuser) BaseDiffusionModel (class in diffusion_models.models.base_diffusion_model) batch_size (TrainingConfiguration attribute) beta_end (LinearBetaScheduler attribute) beta_scheduler (BaseDiffuser attribute) beta_scheduler_config (Checkpoint attribute)

beta_start (LinearBetaScheduler attribute) betas (BaseBetaScheduler attribute) (BetaSchedulerConfiguration attribute) BetaSchedulerConfiguration (class in diffusion_models.utils.schemas) Block (class in diffusion_models.models.SimpleUnet) bnorm1 (Block attribute) bnorm2 (Block attribute)

C

Checkpoint (class in diffusion_models.utils.schemas) checkpoint_path (DiffusionTrainer attribute) checkpoint_rate (TrainingConfiguration attribute) compile() (BaseDiffusionModel method) compute_alpha_bar() (BaseBetaScheduler method) (CosineBetaScheduler method) (LinearBetaScheduler method)

conv0 (SimpleUnet attribute) conv2 (Block attribute) CosineBetaScheduler (class in diffusion_models.gaussian_diffusion.beta_schedulers) current (Timestep attribute)

D

dataloader (DiffusionTrainer attribute) DdimDiffuser (class in diffusion_models.gaussian_diffusion.ddimm_diffuser) denoise() (BaseDiffusionModel method) denoise_batch() (BaseDiffuser method) (DdimDiffuser method) (GaussianDiffuser method) DenoisingMode (class in diffusion_models.gaussian_diffusion.ddimm_diffuser) device (DdimDiffuser attribute) (DiffusionInference attribute) (DiffusionTrainer attribute) (GaussianDiffuser attribute) diffuse() (BaseDiffusionModel method) diffuse_batch() (BaseDiffuser method) (DdimDiffuser method) (GaussianDiffuser method) diffuser (BaseDiffusionModel attribute) diffusion_models module diffusion_models.diffusion_inference module diffusion_models.diffusion_trainer module diffusion_models.gaussian_diffusion module

diffusion_models.gaussian_diffusion.base_diffuser module diffusion_models.gaussian_diffusion.beta_schedulers module diffusion_models.gaussian_diffusion.ddimm_diffuser module diffusion_models.gaussian_diffusion.gaussian_diffuser module diffusion_models.models module diffusion_models.models.base_diffusion_model module diffusion_models.models.SimpleUnet module diffusion_models.utils module diffusion_models.utils.schemas module diffusion_models.utils.tensorboard module DiffusionInference (class in diffusion_models.diffusion_inference) DiffusionTrainer (class in diffusion_models.diffusion_trainer) dim (SinusoidalPositionEmbeddings attribute) downs (SimpleUnet attribute)

E

enforce_zero_terminal_snr() (BaseBetaScheduler method) epoch (Checkpoint attribute) (OldCheckpoint attribute) examples module

examples.infer module examples.train_model module

F

f() (CosineBetaScheduler method) forward() (BaseDiffusionModel method) (Block method) (SimpleUnet method) (SinusoidalPositionEmbeddings method)

from_checkpoint() (DdimDiffuser class method) (GaussianDiffuser class method) from_file() (Checkpoint class method) (OldCheckpoint class method) from_tensors() (BaseBetaScheduler class method)

G

GaussianDiffuser (class in diffusion_models.gaussian_diffusion.gaussian_diffuser) generate() (DiffusionInference method) get_generator() (DiffusionInference method)

get_timestep() (BaseDiffuser method) (DdimDiffuser method) (GaussianDiffuser method) gradient_clip (TrainingConfiguration attribute)

I

image_channels (Checkpoint attribute) (DiffusionInference attribute)

image_rate (LogConfiguration attribute) image_size (DiffusionInference attribute)

L

learning_rate (TrainingConfiguration attribute) Linear (DenoisingMode attribute) LinearBetaScheduler (class in diffusion_models.gaussian_diffusion.beta_schedulers) log_configuration (DiffusionTrainer attribute) log_directory (TensorboardManager attribute) log_images() (TensorboardManager method)

log_metrics() (TensorboardManager method) log_rate (LogConfiguration attribute) log_to_tensorboard() (DiffusionTrainer method) LogConfiguration (class in diffusion_models.utils.schemas) loss (Checkpoint attribute) (OldCheckpoint attribute) loss_function (DiffusionTrainer attribute)

M

max_beta (CosineBetaScheduler attribute) mixed_precision_training (TrainingConfiguration attribute) mode (DdimDiffuser attribute) model (DiffusionInference attribute) (DiffusionTrainer attribute) model_state_dict (Checkpoint attribute) (OldCheckpoint attribute) module diffusion_models diffusion_models.diffusion_inference diffusion_models.diffusion_trainer diffusion_models.gaussian_diffusion diffusion_models.gaussian_diffusion.base_diffuser diffusion_models.gaussian_diffusion.beta_schedulers diffusion_models.gaussian_diffusion.ddimm_diffuser diffusion_models.gaussian_diffusion.gaussian_diffuser diffusion_models.models diffusion_models.models.base_diffusion_model diffusion_models.models.SimpleUnet diffusion_models.utils diffusion_models.utils.schemas diffusion_models.utils.tensorboard examples examples.infer examples.train_model

N

number_of_epochs (TrainingConfiguration attribute)

number_of_images (LogConfiguration attribute) number_of_steps (DdimDiffuser attribute)

O

offset (CosineBetaScheduler attribute) OldCheckpoint (class in diffusion_models.utils.schemas) optimizer (DiffusionTrainer attribute)

optimizer_state_dict (Checkpoint attribute) (OldCheckpoint attribute) output (SimpleUnet attribute)

P

previous (Timestep attribute)

Q

Quadratic (DenoisingMode attribute)

R

relu (Block attribute)

reverse_transforms (DiffusionInference attribute) (DiffusionTrainer attribute)

S

sample_betas() (BaseBetaScheduler method) (CosineBetaScheduler method) (LinearBetaScheduler method) save_checkpoint() (DiffusionTrainer method) scaler (Checkpoint attribute) (DiffusionTrainer attribute) (OldCheckpoint attribute) SimpleUnet (class in diffusion_models.models.SimpleUnet)

SinusoidalPositionEmbeddings (class in diffusion_models.models.SimpleUnet) steps (BaseBetaScheduler attribute) (BaseDiffuser property) (BetaSchedulerConfiguration attribute) (CosineBetaScheduler attribute) (DdimDiffuser property) (GaussianDiffuser property) summary_writer (TensorboardManager attribute)

T

tensorboard_manager (DiffusionTrainer attribute) tensorboard_run_name (Checkpoint attribute) (OldCheckpoint attribute) TensorboardManager (class in diffusion_models.utils.tensorboard) time_mlp (Block attribute) (SimpleUnet attribute) Timestep (class in diffusion_models.utils.schemas) to() (BaseBetaScheduler method) (BaseDiffuser method) (BaseDiffusionModel method) (DdimDiffuser method) (GaussianDiffuser method)

to_file() (Checkpoint method) (OldCheckpoint method) to_new_checkpoint() (OldCheckpoint method) train() (DiffusionTrainer method) training_configuration (DiffusionTrainer attribute) training_name (TrainingConfiguration attribute) TrainingConfiguration (class in diffusion_models.utils.schemas)

U

ups (SimpleUnet attribute)