liopeer
diff --git a/‎_autosummary/diffusion_models.html
+44-11 b/‎_autosummary/diffusion_models.html
+44-11
diff --git a/‎_autosummary/diffusion_models.losses.html
+44-11 b/‎_autosummary/diffusion_models.losses.html
+44-11
diff --git a/‎_autosummary/diffusion_models.losses.kl_divergence.gaussian_kl.html
+47-14 b/‎_autosummary/diffusion_models.losses.kl_divergence.gaussian_kl.html
+47-14
diff --git a/‎_autosummary/diffusion_models.losses.kl_divergence.html
+44-11 b/‎_autosummary/diffusion_models.losses.kl_divergence.html
+44-11
diff --git a/‎_autosummary/diffusion_models.losses.kl_divergence.log_gaussian_kl.html
+47-14 b/‎_autosummary/diffusion_models.losses.kl_divergence.log_gaussian_kl.html
+47-14
diff --git a/‎_autosummary/diffusion_models.models.diffusion.DiffusionModel.html
+476 b/‎_autosummary/diffusion_models.models.diffusion.DiffusionModel.html
+476
diff --git a/‎_autosummary/diffusion_models.models.diffusion.ForwardDiffusion.html
+495 b/‎_autosummary/diffusion_models.models.diffusion.ForwardDiffusion.html
+495
diff --git a/‎_autosummary/diffusion_models.models.diffusion.html
+377 b/‎_autosummary/diffusion_models.models.diffusion.html
+377
diff --git a/‎_autosummary/diffusion_models.models.html
+55-19 b/‎_autosummary/diffusion_models.models.html
+55-19
diff --git a/‎_autosummary/diffusion_models.models.mnist_enc.MNISTEncoder.html
+44-11 b/‎_autosummary/diffusion_models.models.mnist_enc.MNISTEncoder.html
+44-11
diff --git a/‎_autosummary/diffusion_models.models.mnist_enc.html
+47-14 b/‎_autosummary/diffusion_models.models.mnist_enc.html
+47-14
diff --git a/‎_autosummary/diffusion_models.models.multicoil.MultiCoilConv2d.html
+44-11 b/‎_autosummary/diffusion_models.models.multicoil.MultiCoilConv2d.html
+44-11
diff --git a/‎_autosummary/diffusion_models.models.multicoil.MultiCoilReducer.html
+47-14 b/‎_autosummary/diffusion_models.models.multicoil.MultiCoilReducer.html
+47-14
diff --git a/‎_autosummary/diffusion_models.models.multicoil.html
+44-11 b/‎_autosummary/diffusion_models.models.multicoil.html
+44-11
diff --git a/‎_autosummary/diffusion_models.models.positional_encoding.PositionalEncoding.html
+44-11 b/‎_autosummary/diffusion_models.models.positional_encoding.PositionalEncoding.html
+44-11
diff --git a/‎_autosummary/diffusion_models.models.positional_encoding.PositionalEncoding2D.html
+44-11 b/‎_autosummary/diffusion_models.models.positional_encoding.PositionalEncoding2D.html
+44-11
diff --git a/‎_autosummary/diffusion_models.models.positional_encoding.html
+44-11 b/‎_autosummary/diffusion_models.models.positional_encoding.html
+44-11
diff --git a/‎_autosummary/diffusion_models.models.unet.AttentionDecodingBlock.html
+44-11 b/‎_autosummary/diffusion_models.models.unet.AttentionDecodingBlock.html
+44-11
diff --git a/‎_autosummary/diffusion_models.models.unet.AttentionEncodingBlock.html
+44-11 b/‎_autosummary/diffusion_models.models.unet.AttentionEncodingBlock.html
+44-11
diff --git a/‎_autosummary/diffusion_models.models.unet.DecodingBlock.html
+44-11 b/‎_autosummary/diffusion_models.models.unet.DecodingBlock.html
+44-11
diff --git a/‎_autosummary/diffusion_models.models.unet.EncodingBlock.html
+44-11 b/‎_autosummary/diffusion_models.models.unet.EncodingBlock.html
+44-11
diff --git a/‎_autosummary/diffusion_models.models.unet.SelfAttention.html
+44-11 b/‎_autosummary/diffusion_models.models.unet.SelfAttention.html
+44-11
diff --git a/‎_autosummary/diffusion_models.models.unet.UNet.html
+47-14 b/‎_autosummary/diffusion_models.models.unet.UNet.html
+47-14
diff --git a/‎_autosummary/diffusion_models.models.unet.html
+44-11 b/‎_autosummary/diffusion_models.models.unet.html
+44-11
diff --git a/‎_autosummary/diffusion_models.models.vae.ResNet18Decoder.html
+44-11 b/‎_autosummary/diffusion_models.models.vae.ResNet18Decoder.html
+44-11
diff --git a/‎_autosummary/diffusion_models.models.vae.ResNet18Encoder.html
+44-11 b/‎_autosummary/diffusion_models.models.vae.ResNet18Encoder.html
+44-11
diff --git a/‎_autosummary/diffusion_models.models.vae.ResNetBlock.html
+44-11 b/‎_autosummary/diffusion_models.models.vae.ResNetBlock.html
+44-11
diff --git a/‎_autosummary/diffusion_models.models.vae.ResNetDecoderBlock.html
+44-11 b/‎_autosummary/diffusion_models.models.vae.ResNetDecoderBlock.html
+44-11
diff --git a/‎_autosummary/diffusion_models.models.vae.SkipDownSampler.html
+44-11 b/‎_autosummary/diffusion_models.models.vae.SkipDownSampler.html
+44-11
diff --git a/‎_autosummary/diffusion_models.models.vae.SkipUpSampler.html
+44-11 b/‎_autosummary/diffusion_models.models.vae.SkipUpSampler.html
+44-11
diff --git a/‎_autosummary/diffusion_models.models.vae.VariationalAutoencoder.html
+44-11 b/‎_autosummary/diffusion_models.models.vae.VariationalAutoencoder.html
+44-11
diff --git a/‎_autosummary/diffusion_models.models.vae.html
+44-11 b/‎_autosummary/diffusion_models.models.vae.html
+44-11
diff --git a/‎_autosummary/diffusion_models.mri_forward.fft.html
+387 b/‎_autosummary/diffusion_models.mri_forward.fft.html
+387
diff --git a/‎_autosummary/diffusion_models.mri_forward.fft.to_imgspace.html
+393 b/‎_autosummary/diffusion_models.mri_forward.fft.to_imgspace.html
+393
diff --git a/‎_autosummary/diffusion_models.mri_forward.fft.to_kspace.html
+393 b/‎_autosummary/diffusion_models.mri_forward.fft.to_kspace.html
+393
diff --git a/‎_autosummary/diffusion_models.mri_forward.html
+52-16 b/‎_autosummary/diffusion_models.mri_forward.html
+52-16
diff --git a/‎_autosummary/diffusion_models.mri_forward.noise.apply_add_gauss_noise.html
+44-11 b/‎_autosummary/diffusion_models.mri_forward.noise.apply_add_gauss_noise.html
+44-11
diff --git a/‎_autosummary/diffusion_models.mri_forward.noise.apply_mult_gauss_noise.html
+44-11 b/‎_autosummary/diffusion_models.mri_forward.noise.apply_mult_gauss_noise.html
+44-11
diff --git a/‎_autosummary/diffusion_models.mri_forward.noise.html
+47-14 b/‎_autosummary/diffusion_models.mri_forward.noise.html
+47-14
diff --git a/‎_autosummary/diffusion_models.mri_forward.noise.independent_gauss_noise.html
+44-11 b/‎_autosummary/diffusion_models.mri_forward.noise.independent_gauss_noise.html
+44-11
diff --git a/‎_autosummary/diffusion_models.mri_forward.undersampling_mask.StochasticUndersamplingMask.html
+44-11 b/‎_autosummary/diffusion_models.mri_forward.undersampling_mask.StochasticUndersamplingMask.html
+44-11
diff --git a/‎_autosummary/diffusion_models.mri_forward.undersampling_mask.UndersamplingMask.html
+44-11 b/‎_autosummary/diffusion_models.mri_forward.undersampling_mask.UndersamplingMask.html
+44-11
diff --git a/‎_autosummary/diffusion_models.mri_forward.undersampling_mask.gaussian_kernel1d_h.html
+44-11 b/‎_autosummary/diffusion_models.mri_forward.undersampling_mask.gaussian_kernel1d_h.html
+44-11
diff --git a/‎_autosummary/diffusion_models.mri_forward.undersampling_mask.gaussian_kernel1d_v.html
+44-11 b/‎_autosummary/diffusion_models.mri_forward.undersampling_mask.gaussian_kernel1d_v.html
+44-11
diff --git a/‎_autosummary/diffusion_models.mri_forward.undersampling_mask.gaussian_kernel2d.html
+44-11 b/‎_autosummary/diffusion_models.mri_forward.undersampling_mask.gaussian_kernel2d.html
+44-11
diff --git a/‎_autosummary/diffusion_models.mri_forward.undersampling_mask.html
+44-11 b/‎_autosummary/diffusion_models.mri_forward.undersampling_mask.html
+44-11
diff --git a/‎_autosummary/diffusion_models.mri_forward.undersampling_mask.naive_undersampling1d_h.html
+44-11 b/‎_autosummary/diffusion_models.mri_forward.undersampling_mask.naive_undersampling1d_h.html
+44-11
diff --git a/‎_autosummary/diffusion_models.mri_forward.undersampling_mask.naive_undersampling1d_v.html
+44-11 b/‎_autosummary/diffusion_models.mri_forward.undersampling_mask.naive_undersampling1d_v.html
+44-11
diff --git a/‎_autosummary/diffusion_models.mri_forward.undersampling_mask.naive_undersampling2d.html
+44-11 b/‎_autosummary/diffusion_models.mri_forward.undersampling_mask.naive_undersampling2d.html
+44-11
diff --git a/‎_autosummary/diffusion_models.mri_forward.undersampling_mask.normalize_mask.html
+44-11 b/‎_autosummary/diffusion_models.mri_forward.undersampling_mask.normalize_mask.html
+44-11
diff --git a/‎_autosummary/diffusion_models.utils.datasets.Cifar10Dataset.html
+407 b/‎_autosummary/diffusion_models.utils.datasets.Cifar10Dataset.html
+407
diff --git a/‎_autosummary/diffusion_models.utils.datasets.Cifar10DebugDataset.html
+407 b/‎_autosummary/diffusion_models.utils.datasets.Cifar10DebugDataset.html
+407
diff --git a/‎_autosummary/diffusion_models.utils.datasets.FastMRIBrainKSpace.html
+413 b/‎_autosummary/diffusion_models.utils.datasets.FastMRIBrainKSpace.html
+413
diff --git a/‎_autosummary/diffusion_models.utils.datasets.FastMRIBrainKSpaceDebug.html
+413 b/‎_autosummary/diffusion_models.utils.datasets.FastMRIBrainKSpaceDebug.html
+413
diff --git a/‎_autosummary/diffusion_models.utils.datasets.FastMRIBrainTrain.html
+407 b/‎_autosummary/diffusion_models.utils.datasets.FastMRIBrainTrain.html
+407
diff --git a/‎_autosummary/diffusion_models.utils.datasets.FastMRIDebug.html
+407 b/‎_autosummary/diffusion_models.utils.datasets.FastMRIDebug.html
+407
diff --git a/‎_autosummary/diffusion_models.utils.datasets.FastMRIRandCrop.html
+407 b/‎_autosummary/diffusion_models.utils.datasets.FastMRIRandCrop.html
+407
diff --git a/‎_autosummary/diffusion_models.utils.datasets.FastMRIRandCropDebug.html
+407 b/‎_autosummary/diffusion_models.utils.datasets.FastMRIRandCropDebug.html
+407
diff --git a/‎_autosummary/diffusion_models.utils.datasets.ImageNet64Dataset.html
+407 b/‎_autosummary/diffusion_models.utils.datasets.ImageNet64Dataset.html
+407
diff --git a/‎_autosummary/diffusion_models.utils.datasets.MNISTDebugDataset.html
+407 b/‎_autosummary/diffusion_models.utils.datasets.MNISTDebugDataset.html
+407
diff --git a/‎_autosummary/diffusion_models.utils.datasets.MNISTKSpace.html
+407 b/‎_autosummary/diffusion_models.utils.datasets.MNISTKSpace.html
+407
diff --git a/‎_autosummary/diffusion_models.utils.datasets.MNISTKSpaceDebug.html
+407 b/‎_autosummary/diffusion_models.utils.datasets.MNISTKSpaceDebug.html
+407
diff --git a/‎_autosummary/diffusion_models.utils.datasets.MNISTTrainDataset.html
+407 b/‎_autosummary/diffusion_models.utils.datasets.MNISTTrainDataset.html
+407
diff --git a/‎_autosummary/diffusion_models.utils.datasets.QuarterFastMRI.html
+408 b/‎_autosummary/diffusion_models.utils.datasets.QuarterFastMRI.html
+408
diff --git a/‎_autosummary/diffusion_models.utils.datasets.html
+413 b/‎_autosummary/diffusion_models.utils.datasets.html
+413
diff --git a/‎_autosummary/diffusion_models.utils.helpers.bytes_to_gb.html
+44-11 b/‎_autosummary/diffusion_models.utils.helpers.bytes_to_gb.html
+44-11
diff --git a/‎_autosummary/diffusion_models.utils.helpers.complex_to_2channelfloat.html
+44-11 b/‎_autosummary/diffusion_models.utils.helpers.complex_to_2channelfloat.html
+44-11
diff --git a/‎_autosummary/diffusion_models.utils.helpers.dotdict.html
+44-11 b/‎_autosummary/diffusion_models.utils.helpers.dotdict.html
+44-11
diff --git a/‎_autosummary/diffusion_models.utils.helpers.html
+47-14 b/‎_autosummary/diffusion_models.utils.helpers.html
+47-14
diff --git a/‎_autosummary/diffusion_models.utils.html
+55-16 b/‎_autosummary/diffusion_models.utils.html
+55-16
diff --git a/‎_autosummary/diffusion_models.utils.mp_setup.DDP_Proc_Group.html
+54-13 b/‎_autosummary/diffusion_models.utils.mp_setup.DDP_Proc_Group.html
+54-13
diff --git a/‎_autosummary/diffusion_models.utils.mp_setup.html
+44-11 b/‎_autosummary/diffusion_models.utils.mp_setup.html
+44-11
diff --git a/‎_autosummary/diffusion_models.utils.trainer.DiscriminativeTrainer.html
+442 b/‎_autosummary/diffusion_models.utils.trainer.DiscriminativeTrainer.html
+442
diff --git a/‎_autosummary/diffusion_models.utils.trainer.GenerativeTrainer.html
+434 b/‎_autosummary/diffusion_models.utils.trainer.GenerativeTrainer.html
+434
diff --git a/‎_autosummary/diffusion_models.utils.trainer.Trainer.html
+435 b/‎_autosummary/diffusion_models.utils.trainer.Trainer.html
+435
diff --git a/‎_autosummary/diffusion_models.utils.trainer.html
+380 b/‎_autosummary/diffusion_models.utils.trainer.html
+380
diff --git a/‎_modules/diffusion_models/losses/kl_divergence.html
+46-13 b/‎_modules/diffusion_models/losses/kl_divergence.html
+46-13
diff --git a/‎_modules/diffusion_models/models/diffusion.html
+624 b/‎_modules/diffusion_models/models/diffusion.html
+624
diff --git a/‎_modules/diffusion_models/models/mnist_enc.html
+44-11 b/‎_modules/diffusion_models/models/mnist_enc.html
+44-11
diff --git a/‎_modules/diffusion_models/models/multicoil.html
+45-12 b/‎_modules/diffusion_models/models/multicoil.html
+45-12
diff --git a/‎_modules/diffusion_models/models/positional_encoding.html
+44-11 b/‎_modules/diffusion_models/models/positional_encoding.html
+44-11
diff --git a/‎_modules/diffusion_models/models/unet.html
+45-12 b/‎_modules/diffusion_models/models/unet.html
+45-12
diff --git a/‎_modules/diffusion_models/models/vae.html
+45-11 b/‎_modules/diffusion_models/models/vae.html
+45-11
diff --git a/‎_modules/diffusion_models/mri_forward/fft.html
+386 b/‎_modules/diffusion_models/mri_forward/fft.html
+386
diff --git a/‎_modules/diffusion_models/mri_forward/noise.html
+44-11 b/‎_modules/diffusion_models/mri_forward/noise.html
+44-11
diff --git a/‎_modules/diffusion_models/mri_forward/undersampling_mask.html
+44-11 b/‎_modules/diffusion_models/mri_forward/undersampling_mask.html
+44-11
diff --git a/‎_modules/diffusion_models/utils/datasets.html
+589 b/‎_modules/diffusion_models/utils/datasets.html
+589
diff --git a/‎_modules/diffusion_models/utils/helpers.html
+44-11 b/‎_modules/diffusion_models/utils/helpers.html
+44-11
diff --git a/‎_modules/diffusion_models/utils/mp_setup.html
+44-11 b/‎_modules/diffusion_models/utils/mp_setup.html
+44-11
diff --git a/‎_modules/diffusion_models/utils/trainer.html
+684 b/‎_modules/diffusion_models/utils/trainer.html
+684
diff --git a/‎_modules/index.html
+48-11 b/‎_modules/index.html
+48-11
diff --git a/‎_sources/_autosummary/diffusion_models.models.diffusion.DiffusionModel.rst.txt
+29 b/‎_sources/_autosummary/diffusion_models.models.diffusion.DiffusionModel.rst.txt
+29
diff --git a/‎_sources/_autosummary/diffusion_models.models.diffusion.ForwardDiffusion.rst.txt
+27 b/‎_sources/_autosummary/diffusion_models.models.diffusion.ForwardDiffusion.rst.txt
+27
diff --git a/‎_sources/_autosummary/diffusion_models.models.diffusion.rst.txt
+32 b/‎_sources/_autosummary/diffusion_models.models.diffusion.rst.txt
+32
diff --git a/‎_sources/_autosummary/diffusion_models.mri_forward.fft.rst.txt
+31 b/‎_sources/_autosummary/diffusion_models.mri_forward.fft.rst.txt
+31
diff --git a/‎_sources/_autosummary/diffusion_models.mri_forward.fft.to_imgspace.rst.txt
+6 b/‎_sources/_autosummary/diffusion_models.mri_forward.fft.to_imgspace.rst.txt
+6
diff --git a/‎_sources/_autosummary/diffusion_models.mri_forward.fft.to_kspace.rst.txt
+6 b/‎_sources/_autosummary/diffusion_models.mri_forward.fft.to_kspace.rst.txt
+6
diff --git a/‎_sources/_autosummary/diffusion_models.utils.datasets.Cifar10Dataset.rst.txt
+25 b/‎_sources/_autosummary/diffusion_models.utils.datasets.Cifar10Dataset.rst.txt
+25
diff --git a/‎_sources/_autosummary/diffusion_models.utils.datasets.Cifar10DebugDataset.rst.txt
+25 b/‎_sources/_autosummary/diffusion_models.utils.datasets.Cifar10DebugDataset.rst.txt
+25
diff --git a/‎_sources/_autosummary/diffusion_models.utils.datasets.FastMRIBrainKSpace.rst.txt
+27 b/‎_sources/_autosummary/diffusion_models.utils.datasets.FastMRIBrainKSpace.rst.txt
+27
diff --git a/‎_sources/_autosummary/diffusion_models.utils.datasets.FastMRIBrainKSpaceDebug.rst.txt
+27 b/‎_sources/_autosummary/diffusion_models.utils.datasets.FastMRIBrainKSpaceDebug.rst.txt
+27
diff --git a/‎_sources/_autosummary/diffusion_models.utils.datasets.FastMRIBrainTrain.rst.txt
+25 b/‎_sources/_autosummary/diffusion_models.utils.datasets.FastMRIBrainTrain.rst.txt
+25
diff --git a/‎_sources/_autosummary/diffusion_models.utils.datasets.FastMRIDebug.rst.txt
+25 b/‎_sources/_autosummary/diffusion_models.utils.datasets.FastMRIDebug.rst.txt
+25
diff --git a/‎_sources/_autosummary/diffusion_models.utils.datasets.FastMRIRandCrop.rst.txt
+25 b/‎_sources/_autosummary/diffusion_models.utils.datasets.FastMRIRandCrop.rst.txt
+25
diff --git a/‎_sources/_autosummary/diffusion_models.utils.datasets.FastMRIRandCropDebug.rst.txt
+25 b/‎_sources/_autosummary/diffusion_models.utils.datasets.FastMRIRandCropDebug.rst.txt
+25
diff --git a/‎_sources/_autosummary/diffusion_models.utils.datasets.ImageNet64Dataset.rst.txt
+25 b/‎_sources/_autosummary/diffusion_models.utils.datasets.ImageNet64Dataset.rst.txt
+25