Index

_

__init__() (diffusion_models.models.mnist_enc.MNISTEncoder method) (diffusion_models.models.multicoil.MultiCoilConv2d method) (diffusion_models.models.multicoil.MultiCoilReducer method) (diffusion_models.models.nn.CheckpointFunction method) (diffusion_models.models.nn.GroupNorm32 method) (diffusion_models.models.nn.SiLU method) (diffusion_models.models.openai_unet.AttentionBlock method) (diffusion_models.models.openai_unet.Downsample method) (diffusion_models.models.openai_unet.QKVAttention method) (diffusion_models.models.openai_unet.ResBlock method) (diffusion_models.models.openai_unet.SuperResModel method) (diffusion_models.models.openai_unet.TimestepBlock method) (diffusion_models.models.openai_unet.TimestepEmbedSequential method) (diffusion_models.models.openai_unet.UNetModel method) (diffusion_models.models.openai_unet.Upsample method) (diffusion_models.models.positional_encoding.PositionalEncoding method) (diffusion_models.models.positional_encoding.PositionalEncoding2D method) (diffusion_models.models.vae.ResNet18Decoder method) (diffusion_models.models.vae.ResNet18Encoder method) (diffusion_models.models.vae.ResNetBlock method) (diffusion_models.models.vae.ResNetDecoderBlock method) (diffusion_models.models.vae.SkipDownSampler method) (diffusion_models.models.vae.SkipUpSampler method) (diffusion_models.models.vae.VariationalAutoencoder method) (diffusion_models.mri_forward.undersampling_mask.StochasticUndersamplingMask method) (diffusion_models.mri_forward.undersampling_mask.UndersamplingMask method) (diffusion_models.spine_dataset.base_dataset.BaseDataset method) (diffusion_models.spine_dataset.spatial_transformer.SpatialTransformer method) (diffusion_models.utils.helpers.dotdict method) (diffusion_models.utils.mp_setup.DDP_Proc_Group method) (diffusion_models.utils.trainer.DiscriminativeTrainer method) (diffusion_models.utils.trainer.GenerativeTrainer method) (diffusion_models.utils.trainer.Trainer method)

A

apply_add_gauss_noise() (in module diffusion_models.mri_forward.noise) apply_mult_gauss_noise() (in module diffusion_models.mri_forward.noise)

AttentionBlock (class in diffusion_models.models.openai_unet) avg_pool_nd() (in module diffusion_models.models.nn)

B

BaseDataset (class in diffusion_models.spine_dataset.base_dataset)

bytes_to_gb() (in module diffusion_models.utils.helpers)

C

checkpoint() (in module diffusion_models.models.nn) CheckpointFunction (class in diffusion_models.models.nn) clear() (diffusion_models.utils.helpers.dotdict method) collate_fn() (in module diffusion_models.spine_dataset.collations) complex_to_2channelfloat() (in module diffusion_models.utils.helpers) conv_nd() (in module diffusion_models.models.nn) convert_module_to_f16() (in module diffusion_models.models.fp16_util)

convert_module_to_f32() (in module diffusion_models.models.fp16_util) convert_to_fp16() (diffusion_models.models.openai_unet.SuperResModel method) (diffusion_models.models.openai_unet.UNetModel method) convert_to_fp32() (diffusion_models.models.openai_unet.SuperResModel method) (diffusion_models.models.openai_unet.UNetModel method) copy() (diffusion_models.utils.helpers.dotdict method) count_flops() (diffusion_models.models.openai_unet.QKVAttention static method)

D

DDP_Proc_Group (class in diffusion_models.utils.mp_setup) diffusion_models module diffusion_models.losses module diffusion_models.losses.kl_divergence module diffusion_models.models module diffusion_models.models.fp16_util module diffusion_models.models.mnist_enc module diffusion_models.models.multicoil module diffusion_models.models.nn module diffusion_models.models.openai_unet module diffusion_models.models.positional_encoding module diffusion_models.models.vae module diffusion_models.mri_forward module

diffusion_models.mri_forward.fft module diffusion_models.mri_forward.noise module diffusion_models.mri_forward.undersampling_mask module diffusion_models.spine_dataset module diffusion_models.spine_dataset.base_dataset module diffusion_models.spine_dataset.collations module diffusion_models.spine_dataset.spatial_transformer module diffusion_models.utils module diffusion_models.utils.helpers module diffusion_models.utils.mp_setup module diffusion_models.utils.trainer module DiscriminativeTrainer (class in diffusion_models.utils.trainer) dotdict (class in diffusion_models.utils.helpers) Downsample (class in diffusion_models.models.openai_unet)

F

forward() (diffusion_models.models.openai_unet.QKVAttention method) (diffusion_models.models.openai_unet.ResBlock method) (diffusion_models.models.openai_unet.SuperResModel method) (diffusion_models.models.openai_unet.TimestepBlock method) (diffusion_models.models.openai_unet.TimestepEmbedSequential method) (diffusion_models.models.openai_unet.UNetModel method) (diffusion_models.models.positional_encoding.PositionalEncoding method) (diffusion_models.spine_dataset.spatial_transformer.SpatialTransformer method)

fromkeys() (diffusion_models.utils.helpers.dotdict method)

G

gauss_gen3D() (in module diffusion_models.spine_dataset.spatial_transformer) gaussian_kernel1d_h() (in module diffusion_models.mri_forward.undersampling_mask) gaussian_kernel1d_v() (in module diffusion_models.mri_forward.undersampling_mask) gaussian_kernel2d() (in module diffusion_models.mri_forward.undersampling_mask) gaussian_kl() (in module diffusion_models.losses.kl_divergence)

GenerativeTrainer (class in diffusion_models.utils.trainer) get() (diffusion_models.utils.helpers.dotdict method) get_feature_vectors() (diffusion_models.models.openai_unet.SuperResModel method) (diffusion_models.models.openai_unet.UNetModel method) get_pos_encoding() (diffusion_models.models.positional_encoding.PositionalEncoding method) GroupNorm32 (class in diffusion_models.models.nn)

I

independent_gauss_noise() (in module diffusion_models.mri_forward.noise) inner_dtype (diffusion_models.models.openai_unet.SuperResModel property) (diffusion_models.models.openai_unet.UNetModel property)

items() (diffusion_models.utils.helpers.dotdict method)

K

keys() (diffusion_models.utils.helpers.dotdict method)

L

linear() (in module diffusion_models.models.nn)

log_gaussian_kl() (in module diffusion_models.losses.kl_divergence)

M

make_master_params() (in module diffusion_models.models.fp16_util) master_params_to_model_params() (in module diffusion_models.models.fp16_util) mean_flat() (in module diffusion_models.models.nn) MNISTEncoder (class in diffusion_models.models.mnist_enc) model_grads_to_master_grads() (in module diffusion_models.models.fp16_util) module diffusion_models diffusion_models.losses diffusion_models.losses.kl_divergence diffusion_models.models diffusion_models.models.fp16_util diffusion_models.models.mnist_enc diffusion_models.models.multicoil diffusion_models.models.nn diffusion_models.models.openai_unet diffusion_models.models.positional_encoding diffusion_models.models.vae diffusion_models.mri_forward diffusion_models.mri_forward.fft diffusion_models.mri_forward.noise diffusion_models.mri_forward.undersampling_mask diffusion_models.spine_dataset diffusion_models.spine_dataset.base_dataset diffusion_models.spine_dataset.collations diffusion_models.spine_dataset.spatial_transformer diffusion_models.utils diffusion_models.utils.helpers diffusion_models.utils.mp_setup diffusion_models.utils.trainer

MultiCoilConv2d (class in diffusion_models.models.multicoil) MultiCoilReducer (class in diffusion_models.models.multicoil)

N

naive_undersampling1d_h() (in module diffusion_models.mri_forward.undersampling_mask) naive_undersampling1d_v() (in module diffusion_models.mri_forward.undersampling_mask)

naive_undersampling2d() (in module diffusion_models.mri_forward.undersampling_mask) normalization() (in module diffusion_models.models.nn) normalize_mask() (in module diffusion_models.mri_forward.undersampling_mask)

P

pop() (diffusion_models.utils.helpers.dotdict method) popitem() (diffusion_models.utils.helpers.dotdict method)

PositionalEncoding (class in diffusion_models.models.positional_encoding) PositionalEncoding2D (class in diffusion_models.models.positional_encoding)

Q

QKVAttention (class in diffusion_models.models.openai_unet)

R

ResBlock (class in diffusion_models.models.openai_unet) ResNet18Decoder (class in diffusion_models.models.vae)

ResNet18Encoder (class in diffusion_models.models.vae) ResNetBlock (class in diffusion_models.models.vae) ResNetDecoderBlock (class in diffusion_models.models.vae)

S

scale_module() (in module diffusion_models.models.nn) setdefault() (diffusion_models.utils.helpers.dotdict method) SiLU (class in diffusion_models.models.nn) SkipDownSampler (class in diffusion_models.models.vae)

SkipUpSampler (class in diffusion_models.models.vae) SpatialTransformer (class in diffusion_models.spine_dataset.spatial_transformer) StochasticUndersamplingMask (class in diffusion_models.mri_forward.undersampling_mask) SuperResModel (class in diffusion_models.models.openai_unet)

T

timestep_embedding() (in module diffusion_models.models.nn) TimestepBlock (class in diffusion_models.models.openai_unet) TimestepEmbedSequential (class in diffusion_models.models.openai_unet) to_imgspace() (in module diffusion_models.mri_forward.fft)

to_kspace() (in module diffusion_models.mri_forward.fft) train() (diffusion_models.utils.trainer.DiscriminativeTrainer method) (diffusion_models.utils.trainer.GenerativeTrainer method) (diffusion_models.utils.trainer.Trainer method) Trainer (class in diffusion_models.utils.trainer)

U

UndersamplingMask (class in diffusion_models.mri_forward.undersampling_mask) UNetModel (class in diffusion_models.models.openai_unet) unflatten_master_params() (in module diffusion_models.models.fp16_util)

update() (diffusion_models.utils.helpers.dotdict method) update_ema() (in module diffusion_models.models.nn) Upsample (class in diffusion_models.models.openai_unet)

V

values() (diffusion_models.utils.helpers.dotdict method)

VariationalAutoencoder (class in diffusion_models.models.vae)

Z

zero_grad() (in module diffusion_models.models.fp16_util)

zero_module() (in module diffusion_models.models.nn)