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) batch_size (TrainingConfiguration attribute) benchmark (DiffusionTrainer 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 SimpleUnet) bnorm1 (Block attribute) bnorm2 (Block attribute)

C

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

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

D

dataloader (DiffusionTrainer attribute) denoise_batch() (BaseDiffuser method) (GaussianDiffuser method) device (DiffusionTrainer attribute) (GaussianDiffuser attribute) diffuse_batch() (BaseDiffuser method) (GaussianDiffuser method) diffusion_models 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.gaussian_diffuser module diffusion_models.utils module diffusion_models.utils.schemas module diffusion_models.utils.tensorboard module DiffusionTrainer (class in diffusion_models.diffusion_trainer) dim (SinusoidalPositionEmbeddings attribute) down_channels (SimpleUnet 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() (Block method) (SimpleUnet method) (SinusoidalPositionEmbeddings method)

from_checkpoint() (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)

gradient_clip (TrainingConfiguration attribute)

I

image_channels (Checkpoint attribute) (SimpleUnet attribute)

image_rate (LogConfiguration attribute) (TrainingConfiguration attribute)

L

learning_rate (TrainingConfiguration 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) (TrainingConfiguration 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) model (DiffusionTrainer attribute) model_state_dict (Checkpoint attribute) (OldCheckpoint attribute) module diffusion_models diffusion_models.diffusion_trainer diffusion_models.gaussian_diffusion diffusion_models.gaussian_diffusion.base_diffuser diffusion_models.gaussian_diffusion.beta_schedulers diffusion_models.gaussian_diffusion.gaussian_diffuser diffusion_models.utils diffusion_models.utils.schemas diffusion_models.utils.tensorboard examples examples.infer examples.train_model SimpleUnet

N

number_of_epochs (TrainingConfiguration attribute)

number_of_images (LogConfiguration attribute)

O

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

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

R

relu (Block attribute)

reverse_transforms (DiffusionTrainer attribute)

S

sample_betas() (BaseBetaScheduler method) (CosineBetaScheduler method) (LinearBetaScheduler method) save_checkpoint() (DiffusionTrainer method) scaler (Checkpoint attribute) (DiffusionTrainer attribute) (OldCheckpoint attribute) scheduler (DiffusionTrainer attribute)

SimpleUnet module SimpleUnet (class in SimpleUnet) SinusoidalPositionEmbeddings (class in SimpleUnet) steps (BaseBetaScheduler attribute) (BetaSchedulerConfiguration attribute) (CosineBetaScheduler attribute) summary_writer (TensorboardManager attribute)

T

tensorboard_manager (DiffusionTrainer attribute) tensorboard_run_name (Checkpoint attribute) (OldCheckpoint attribute) TensorboardManager (class in diffusion_models.utils.tensorboard) time_emb_dim (SimpleUnet attribute) time_mlp (Block attribute) (SimpleUnet attribute) to() (BaseBetaScheduler method) (BaseDiffuser 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

up_channels (SimpleUnet attribute)

ups (SimpleUnet attribute)