Hide navigation sidebar

Hide table of contents sidebar

Skip to content

Toggle site navigation sidebar

diffusionmodels 0.0.1 documentation

Toggle table of contents sidebar

diffusionmodels 0.0.1 documentation

diffusion_models
Toggle navigation of diffusion_models
- diffusion_models.losses
  Toggle navigation of diffusion_models.losses
  - diffusion_models.losses.kl_divergence
    Toggle navigation of diffusion_models.losses.kl_divergence
    - diffusion_models.losses.kl_divergence.gaussian_kl
    - diffusion_models.losses.kl_divergence.log_gaussian_kl
- diffusion_models.models
  Toggle navigation of diffusion_models.models
  - diffusion_models.models.fp16_util
    Toggle navigation of diffusion_models.models.fp16_util
  - diffusion_models.models.mnist_enc
    Toggle navigation of diffusion_models.models.mnist_enc
    - diffusion_models.models.mnist_enc.MNISTEncoder
  - diffusion_models.models.multicoil
    Toggle navigation of diffusion_models.models.multicoil
    - diffusion_models.models.multicoil.MultiCoilConv2d
    - diffusion_models.models.multicoil.MultiCoilReducer
  - diffusion_models.models.nn
    Toggle navigation of diffusion_models.models.nn
  - diffusion_models.models.openai_unet
    Toggle navigation of diffusion_models.models.openai_unet
  - diffusion_models.models.positional_encoding
    Toggle navigation of diffusion_models.models.positional_encoding
    - diffusion_models.models.positional_encoding.PositionalEncoding
    - diffusion_models.models.positional_encoding.PositionalEncoding2D
  - diffusion_models.models.vae
    Toggle navigation of diffusion_models.models.vae
- diffusion_models.mri_forward
  Toggle navigation of diffusion_models.mri_forward
  - diffusion_models.mri_forward.fft
    Toggle navigation of diffusion_models.mri_forward.fft
    - diffusion_models.mri_forward.fft.to_imgspace
    - diffusion_models.mri_forward.fft.to_kspace
  - diffusion_models.mri_forward.noise
    Toggle navigation of diffusion_models.mri_forward.noise
  - diffusion_models.mri_forward.undersampling_mask
    Toggle navigation of diffusion_models.mri_forward.undersampling_mask
- diffusion_models.spine_dataset
  Toggle navigation of diffusion_models.spine_dataset
  - diffusion_models.spine_dataset.base_dataset
    Toggle navigation of diffusion_models.spine_dataset.base_dataset
    - diffusion_models.spine_dataset.base_dataset.BaseDataset
  - diffusion_models.spine_dataset.collations
    Toggle navigation of diffusion_models.spine_dataset.collations
    - diffusion_models.spine_dataset.collations.collate_fn
  - diffusion_models.spine_dataset.spatial_transformer
    Toggle navigation of diffusion_models.spine_dataset.spatial_transformer
    - diffusion_models.spine_dataset.spatial_transformer.gauss_gen3D
    - diffusion_models.spine_dataset.spatial_transformer.SpatialTransformer
- diffusion_models.utils
  Toggle navigation of diffusion_models.utils
  - diffusion_models.utils.helpers
    Toggle navigation of diffusion_models.utils.helpers
  - diffusion_models.utils.mp_setup
    Toggle navigation of diffusion_models.utils.mp_setup
    - diffusion_models.utils.mp_setup.DDP_Proc_Group
  - diffusion_models.utils.trainer
    Toggle navigation of diffusion_models.utils.trainer

Toggle table of contents sidebar

diffusion_models.models.fp16_util.zero_grad¶

diffusion_models.models.fp16_util.zero_grad(model_params)[source]¶

diffusion_models.models.mnist_enc

diffusion_models.models.fp16_util.unflatten_master_params

Copyright © 2024, Lionel Peer

Made with Sphinx and @pradyunsg's Furo

On this page

diffusion_models.models.fp16_util.zero_grad
- zero_grad()