Index _ | A | B | C | D | E | F | G | H | I | K | L | M | N | O | P | Q | R | S | T | U | V | W _ __call__() (vision_architectures.transforms.clipping.Clip method) (vision_architectures.transforms.clipping.Clipd method) (vision_architectures.transforms.croppad.RandSpatialCropSamplesWithCropTracking method) (vision_architectures.utils.activation_checkpointing.ActivationCheckpointing method) (vision_architectures.utils.splitter_merger.Splitter method) __init__() (vision_architectures.blocks.cnn.CNNBlock2D method) (vision_architectures.blocks.cnn.CNNBlock3D method) (vision_architectures.blocks.cnn.MultiResCNNBlock2D method) (vision_architectures.blocks.cnn.MultiResCNNBlock3D method) (vision_architectures.blocks.cnn.TensorSplittingConv method) (vision_architectures.blocks.heads_3d.ClassificationHead3D method) (vision_architectures.blocks.heads_3d.SegmentationHead3D method) (vision_architectures.blocks.mbconv_3d.MBConv3D method) (vision_architectures.blocks.se.SEBlock3D method) (vision_architectures.blocks.transformer.Attention1DMLP method) (vision_architectures.blocks.transformer.Attention1DWithMLP method) (vision_architectures.blocks.transformer.Attention3DMLP method) (vision_architectures.blocks.transformer.Attention3DWithMLP method) (vision_architectures.blocks.transformer.TransformerDecoderBlock1D method) (vision_architectures.blocks.transformer.TransformerDecoderBlock3D method) (vision_architectures.image_readers.safetensors_reader.SafetensorsReader method) (vision_architectures.layers.attention.Attention1D method) (vision_architectures.layers.attention.Attention3D method) (vision_architectures.layers.codebook.Codebook method) (vision_architectures.layers.embeddings.AbsolutePositionEmbeddings1D method) (vision_architectures.layers.embeddings.AbsolutePositionEmbeddings3D method) (vision_architectures.layers.embeddings.PatchEmbeddings3D method) (vision_architectures.layers.embeddings.RelativePositionEmbeddings3D method) (vision_architectures.layers.embeddings.RelativePositionEmbeddings3DMetaNetwork method) (vision_architectures.layers.latent_space.GaussianLatentSpace method) (vision_architectures.layers.latent_space.LatentDecoder method) (vision_architectures.layers.latent_space.LatentEncoder method) (vision_architectures.layers.scale.PixelShuffleDownsample3D method) (vision_architectures.layers.scale.PixelShuffleUpsample3D method) (vision_architectures.nets.fpn_2d.FPN2D method) (vision_architectures.nets.fpn_2d.FPN2DBlock method) (vision_architectures.nets.fpn_3d.FPN3D method) (vision_architectures.nets.fpn_3d.FPN3DBlock method) (vision_architectures.nets.maxvit_3d.MaxViT3DBlock method) (vision_architectures.nets.maxvit_3d.MaxViT3DEncoder method) (vision_architectures.nets.maxvit_3d.MaxViT3DStem method) (vision_architectures.nets.maxvit_3d.MaxViT3DStem0 method) (vision_architectures.nets.perceiver_3d.Perceiver3DChannelMapping method) (vision_architectures.nets.perceiver_3d.Perceiver3DDecoder method) (vision_architectures.nets.perceiver_3d.Perceiver3DEncoder method) (vision_architectures.nets.perceiver_3d.Perceiver3DEncoderEncode method) (vision_architectures.nets.perceiver_3d.Perceiver3DEncoderProcess method) (vision_architectures.nets.swin_3d.Swin3DBlock method) (vision_architectures.nets.swin_3d.Swin3DEncoder method) (vision_architectures.nets.swin_3d.Swin3DLayer method) (vision_architectures.nets.swin_3d.Swin3DMIM method) (vision_architectures.nets.swin_3d.Swin3DMIMDecoder method) (vision_architectures.nets.swin_3d.Swin3DModel method) (vision_architectures.nets.swin_3d.Swin3DPatchMerging method) (vision_architectures.nets.swin_3d.Swin3DStage method) (vision_architectures.nets.swinv2_3d.SwinV23DBlock method) (vision_architectures.nets.swinv2_3d.SwinV23DDecoder method) (vision_architectures.nets.swinv2_3d.SwinV23DEncoder method) (vision_architectures.nets.swinv2_3d.SwinV23DLayer method) (vision_architectures.nets.swinv2_3d.SwinV23DLayerLogitScale method) (vision_architectures.nets.swinv2_3d.SwinV23DMIM method) (vision_architectures.nets.swinv2_3d.SwinV23DModel method) (vision_architectures.nets.swinv2_3d.SwinV23DPatchMerging method) (vision_architectures.nets.swinv2_3d.SwinV23DPatchSplitting method) (vision_architectures.nets.swinv2_3d.SwinV23DReconstructionDecoder method) (vision_architectures.nets.swinv2_3d.SwinV23DSimMIM method) (vision_architectures.nets.swinv2_3d.SwinV23DStage method) (vision_architectures.nets.swinv2_3d.SwinV23DVAEMIM method) (vision_architectures.nets.symswin_3d.SymSwin3DBlock method) (vision_architectures.nets.symswin_3d.SymSwin3DEmbeddings method) (vision_architectures.nets.symswin_3d.SymSwin3DEncoder method) (vision_architectures.nets.symswin_3d.SymSwin3DLayer method) (vision_architectures.nets.symswin_3d.SymSwin3DLayerMLP method) (vision_architectures.nets.symswin_3d.SymSwin3DMHSA method) (vision_architectures.nets.symswin_3d.SymSwin3DMIM method) (vision_architectures.nets.symswin_3d.SymSwin3DModel method) (vision_architectures.nets.symswin_3d.SymSwin3DPatchEmbeddings method) (vision_architectures.nets.symswin_3d.SymSwin3DPatchMerging method) (vision_architectures.nets.symswin_3d.SymSwin3DReconstructionDecoder method) (vision_architectures.nets.symswin_3d.SymSwin3DSimMIM method) (vision_architectures.nets.symswin_3d.SymSwin3DStage method) (vision_architectures.nets.symswin_3d.SymSwin3DVAEMIM method) (vision_architectures.nets.unetr_3d_decoder.UNetR3DBlock method) (vision_architectures.nets.unetr_3d_decoder.UNetR3DConvBlock method) (vision_architectures.nets.unetr_3d_decoder.UNetR3DDecoder method) (vision_architectures.nets.unetr_3d_decoder.UNetR3DDeConvBlock method) (vision_architectures.nets.upernet_2d.UPerNet2D method) (vision_architectures.nets.upernet_2d.UPerNet2DFusion method) (vision_architectures.nets.upernet_3d.UPerNet3D method) (vision_architectures.nets.upernet_3d.UPerNet3DFusion method) (vision_architectures.nets.vit_3d.ViT3DDecoder method) (vision_architectures.nets.vit_3d.ViT3DEncoder method) (vision_architectures.nets.vit_3d.ViT3DMIM method) (vision_architectures.nets.vit_3d.ViT3DMIMDecoder method) (vision_architectures.nets.vit_3d.ViT3DModel method) (vision_architectures.nets.vit_3d.ViT3DSimMIM method) (vision_architectures.schedulers.cyclic.CyclicAnnealingLR method) (vision_architectures.schedulers.cyclic.CyclicAnnealingScheduler method) (vision_architectures.schedulers.cyclic.SineLR method) (vision_architectures.schedulers.cyclic.SineScheduler method) (vision_architectures.schedulers.lrs.ConstantLRWithWarmup method) (vision_architectures.schedulers.noise.CosineNoiseScheduler method) (vision_architectures.schedulers.noise.ExponentialNoiseScheduler method) (vision_architectures.schedulers.noise.FibonacciNoiseScheduler method) (vision_architectures.schedulers.noise.LinearNoiseScheduler method) (vision_architectures.schedulers.noise.NoiseScheduler method) (vision_architectures.schedulers.noise.SigmoidNoiseScheduler method) (vision_architectures.schedulers.noise.SquareRootNoiseScheduler method) (vision_architectures.schedulers.sigmoid.SigmoidLR method) (vision_architectures.schedulers.sigmoid.SigmoidScheduler method) (vision_architectures.transforms.clipping.Clip method) (vision_architectures.transforms.clipping.Clipd method) (vision_architectures.transforms.croppad.CropForegroundWithCropTrackingd method) (vision_architectures.transforms.croppad.RandSpatialCropSamplesWithCropTracking method) (vision_architectures.transforms.croppad.RandSpatialCropSamplesWithCropTrackingd method) (vision_architectures.transforms.spatial.RandFlipWithCropTrackingd method) (vision_architectures.transforms.spatial.ResizedWithCropTrackingd method) (vision_architectures.utils.activation_checkpointing.ActivationCheckpointing method) (vision_architectures.utils.normalizations.DyT method) (vision_architectures.utils.pipeline_parallelism.PipelineModule method) (vision_architectures.utils.residuals.StochasticDepthResidual method) (vision_architectures.utils.splitter_merger.Merger method) (vision_architectures.utils.splitter_merger.Splitter method) (vision_architectures.utils.timesteps.TimestepSampler method) A AbsolutePositionEmbeddings1D (class in vision_architectures.layers.embeddings) AbsolutePositionEmbeddings3D (class in vision_architectures.layers.embeddings) activation (vision_architectures.blocks.cnn.CNNBlockConfig attribute) (vision_architectures.blocks.mbconv_3d.MBConv3DConfig attribute) (vision_architectures.blocks.se.SEBlock3DConfig attribute) (vision_architectures.blocks.transformer.Attention1DMLPConfig attribute) activation_kwargs (vision_architectures.blocks.cnn.CNNBlockConfig attribute) ActivationCheckpointing (class in vision_architectures.utils.activation_checkpointing) add_noise() (vision_architectures.schedulers.noise.NoiseScheduler method) add_stochastic_depth_dropout() (in module vision_architectures.utils.residuals) add_tsp_to_module() (in module vision_architectures.blocks.cnn) Attention1D (class in vision_architectures.layers.attention) Attention1DMLP (class in vision_architectures.blocks.transformer) Attention1DWithMLP (class in vision_architectures.blocks.transformer) Attention3D (class in vision_architectures.layers.attention) Attention3DMLP (class in vision_architectures.blocks.transformer) Attention3DWithMLP (class in vision_architectures.blocks.transformer) attn_drop_prob (vision_architectures.layers.attention.Attention1DConfig attribute) (vision_architectures.nets.vit_3d.ViT3DDecoderConfig attribute) (vision_architectures.nets.vit_3d.ViT3DEncoderConfig attribute) B blocks (vision_architectures.nets.fpn_2d.FPN2DConfig attribute) (vision_architectures.nets.fpn_3d.FPN3DConfig attribute) BOTTOM (vision_architectures.schedulers.cyclic.Phase attribute) C calculate_losses() (vision_architectures.layers.codebook.Codebook method) calculate_perplexity() (vision_architectures.layers.codebook.Codebook method) calculate_relative_position_bias() (vision_architectures.nets.symswin_3d.SymSwin3DMHSA method) ceil_softplus_clamp() (in module vision_architectures.utils.clamping) ClassificationHead3D (class in vision_architectures.blocks.heads_3d) Clip (class in vision_architectures.transforms.clipping) Clipd (class in vision_architectures.transforms.clipping) CNNBlock2D (class in vision_architectures.blocks.cnn) CNNBlock3D (class in vision_architectures.blocks.cnn) Codebook (class in vision_architectures.layers.codebook) concat_features() (vision_architectures.nets.upernet_2d.UPerNet2DFusion method) (vision_architectures.nets.upernet_3d.UPerNet3DFusion method) ConstantLRWithWarmup (class in vision_architectures.schedulers.lrs) conv_kernel_size (vision_architectures.nets.unetr_3d_decoder.UNetR3DDecoderConfig attribute) conv_kwargs (vision_architectures.blocks.cnn.CNNBlockConfig attribute) CosineNoiseScheduler (class in vision_architectures.schedulers.noise) CropForegroundWithCropTrackingd (class in vision_architectures.transforms.croppad) cross_entropy_loss_fn() (vision_architectures.nets.unetr_3d_decoder.UNetR3DDecoder static method) CyclicAnnealingLR (class in vision_architectures.schedulers.cyclic) CyclicAnnealingScheduler (class in vision_architectures.schedulers.cyclic) D decode (vision_architectures.nets.perceiver_3d.Perceiver3DConfig attribute) decoder (vision_architectures.nets.unetr_3d_decoder.UNetR3DConfig attribute) decoder_depth (vision_architectures.nets.vit_3d.ViT3DDecoderConfig attribute) depth (vision_architectures.nets.maxvit_3d.MaxViT3DStem0Config attribute) (vision_architectures.nets.maxvit_3d.MaxViT3DStemConfig attribute) (vision_architectures.nets.swin_3d.Swin3DStageConfig attribute) (vision_architectures.nets.swinv2_3d.SwinV23DStageConfig attribute) dim (vision_architectures.blocks.mbconv_3d.MBConv3DConfig attribute) (vision_architectures.blocks.se.SEBlock3DConfig attribute) (vision_architectures.blocks.transformer.Attention1DMLPConfig attribute) (vision_architectures.blocks.transformer.Attention1DWithMLPConfig attribute) (vision_architectures.blocks.transformer.Attention3DWithMLPConfig attribute) (vision_architectures.layers.attention.Attention1DConfig attribute) (vision_architectures.layers.codebook.CodebookConfig attribute) (vision_architectures.layers.embeddings.AbsolutePositionEmbeddings1DConfig attribute) (vision_architectures.layers.embeddings.AbsolutePositionEmbeddings3DConfig attribute) (vision_architectures.layers.embeddings.PatchEmbeddings3DConfig attribute) (vision_architectures.layers.latent_space.LatentDecoderConfig property) (vision_architectures.layers.latent_space.LatentEncoderConfig property) (vision_architectures.nets.fpn_2d.FPN2DBlockConfig attribute) (vision_architectures.nets.fpn_2d.FPN2DConfig property) (vision_architectures.nets.fpn_3d.FPN3DBlockConfig attribute) (vision_architectures.nets.fpn_3d.FPN3DConfig property) (vision_architectures.nets.maxvit_3d.MaxViT3DStem0Config attribute) (vision_architectures.nets.perceiver_3d.Perceiver3DDecoderConfig attribute) (vision_architectures.nets.perceiver_3d.Perceiver3DEncoderConfig property) (vision_architectures.nets.perceiver_3d.Perceiver3DEncoderEncodeConfig attribute) (vision_architectures.nets.perceiver_3d.Perceiver3DEncoderProcessConfig attribute) (vision_architectures.nets.swin_3d.Swin3DConfig attribute) (vision_architectures.nets.swin_3d.Swin3DStageConfig attribute) (vision_architectures.nets.swinv2_3d.SwinV23DDecoderConfig attribute) (vision_architectures.nets.swinv2_3d.SwinV23DStageConfig attribute) (vision_architectures.nets.upernet_2d.UPerNet2DFusionConfig attribute) (vision_architectures.nets.upernet_3d.UPerNet3DFusionConfig attribute) (vision_architectures.nets.vit_3d.ViT3DDecoderConfig attribute) (vision_architectures.nets.vit_3d.ViT3DEncoderConfig attribute) dim_qk (vision_architectures.layers.attention.Attention1DConfig property) dim_v (vision_architectures.layers.attention.Attention1DConfig property) DOWN (vision_architectures.schedulers.cyclic.Phase attribute) drop_prob (vision_architectures.blocks.cnn.CNNBlockConfig attribute) (vision_architectures.nets.swin_3d.Swin3DConfig attribute) (vision_architectures.nets.swinv2_3d.SwinV23DDecoderConfig attribute) (vision_architectures.nets.vit_3d.ViT3DConfig attribute) DyT (class in vision_architectures.utils.normalizations) E ema_decay (vision_architectures.layers.codebook.CodebookConfig attribute) embed_spacing_info (vision_architectures.nets.swin_3d.Swin3DConfig attribute) (vision_architectures.nets.swinv2_3d.SwinV23DDecoderConfig attribute) embed_spacings_in_position_embeddings() (in module vision_architectures.nets.symswin_3d) enabled_outputs (vision_architectures.nets.upernet_2d.UPerNet2DConfig attribute) (vision_architectures.nets.upernet_3d.UPerNet3DConfig attribute) encode (vision_architectures.nets.perceiver_3d.Perceiver3DEncoderConfig attribute) encoder_depth (vision_architectures.nets.vit_3d.ViT3DEncoderConfig attribute) expand() (vision_architectures.utils.splitter_merger.Splitter method) expansion_ratio (vision_architectures.blocks.mbconv_3d.MBConv3DConfig attribute) ExponentialNoiseScheduler (class in vision_architectures.schedulers.noise) extend_mode (vision_architectures.utils.splitter_merger.SplitterConfig attribute) extra_repr() (vision_architectures.blocks.cnn.TensorSplittingConv method) (vision_architectures.utils.activation_checkpointing.ActivationCheckpointing method) (vision_architectures.utils.normalizations.DyT method) (vision_architectures.utils.pipeline_parallelism.PipelineModule method) (vision_architectures.utils.residuals.StochasticDepthResidual method) F FibonacciNoiseScheduler (class in vision_architectures.schedulers.noise) filter_ratios (vision_architectures.blocks.cnn.MultiResCNNBlockConfig attribute) final_layer_kernel_size (vision_architectures.nets.unetr_3d_decoder.UNetR3DDecoderConfig attribute) final_window_size (vision_architectures.nets.swin_3d.Swin3DPatchSplittingConfig attribute) (vision_architectures.nets.swinv2_3d.SwinV23DPatchSplittingConfig attribute) floor_softplus_clamp() (in module vision_architectures.utils.clamping) fold_back_3d() (in module vision_architectures.nets.perceiver_3d) forward() (vision_architectures.blocks.cnn.TensorSplittingConv method) (vision_architectures.blocks.mbconv_3d.MBConv3D method) (vision_architectures.blocks.se.SEBlock3D method) (vision_architectures.blocks.transformer.Attention1DMLP method) (vision_architectures.blocks.transformer.Attention1DWithMLP method) (vision_architectures.blocks.transformer.Attention3DMLP method) (vision_architectures.blocks.transformer.Attention3DWithMLP method) (vision_architectures.blocks.transformer.TransformerDecoderBlock1D method) (vision_architectures.blocks.transformer.TransformerDecoderBlock3D method) (vision_architectures.blocks.transformer.TransformerEncoderBlock1D method) (vision_architectures.blocks.transformer.TransformerEncoderBlock3D method) (vision_architectures.layers.attention.Attention1D method) (vision_architectures.layers.attention.Attention3D method) (vision_architectures.layers.codebook.Codebook method) (vision_architectures.layers.embeddings.AbsolutePositionEmbeddings1D method) (vision_architectures.layers.embeddings.AbsolutePositionEmbeddings3D method) (vision_architectures.layers.embeddings.RelativePositionEmbeddings3D method) (vision_architectures.layers.embeddings.RelativePositionEmbeddings3DMetaNetwork method) (vision_architectures.layers.latent_space.GaussianLatentSpace method) (vision_architectures.layers.latent_space.LatentDecoder method) (vision_architectures.layers.latent_space.LatentEncoder method) (vision_architectures.layers.scale.PixelShuffleDownsample3D method) (vision_architectures.layers.scale.PixelShuffleUpsample3D method) (vision_architectures.nets.fpn_2d.FPN2D method) (vision_architectures.nets.fpn_2d.FPN2DBlock method) (vision_architectures.nets.fpn_3d.FPN3D method) (vision_architectures.nets.fpn_3d.FPN3DBlock method) (vision_architectures.nets.maxvit_3d.MaxViT3DBlock method) (vision_architectures.nets.maxvit_3d.MaxViT3DEncoder method) (vision_architectures.nets.maxvit_3d.MaxViT3DStem method) (vision_architectures.nets.maxvit_3d.MaxViT3DStem0 method) (vision_architectures.nets.perceiver_3d.Perceiver3DChannelMapping method) (vision_architectures.nets.perceiver_3d.Perceiver3DDecoder method) (vision_architectures.nets.perceiver_3d.Perceiver3DEncoder method) (vision_architectures.nets.perceiver_3d.Perceiver3DEncoderEncode method) (vision_architectures.nets.perceiver_3d.Perceiver3DEncoderProcess method) (vision_architectures.nets.swin_3d.Swin3DBlock method) (vision_architectures.nets.swin_3d.Swin3DEncoder method) (vision_architectures.nets.swin_3d.Swin3DLayer method) (vision_architectures.nets.swin_3d.Swin3DMIM method) (vision_architectures.nets.swin_3d.Swin3DMIMDecoder method) (vision_architectures.nets.swin_3d.Swin3DModel method) (vision_architectures.nets.swin_3d.Swin3DPatchMerging method) (vision_architectures.nets.swin_3d.Swin3DStage method) (vision_architectures.nets.swinv2_3d.SwinV23DBlock method) (vision_architectures.nets.swinv2_3d.SwinV23DDecoder method) (vision_architectures.nets.swinv2_3d.SwinV23DEncoder method) (vision_architectures.nets.swinv2_3d.SwinV23DLayer method) (vision_architectures.nets.swinv2_3d.SwinV23DLayerLogitScale method) (vision_architectures.nets.swinv2_3d.SwinV23DModel method) (vision_architectures.nets.swinv2_3d.SwinV23DPatchMerging method) (vision_architectures.nets.swinv2_3d.SwinV23DPatchSplitting method) (vision_architectures.nets.swinv2_3d.SwinV23DReconstructionDecoder method) (vision_architectures.nets.swinv2_3d.SwinV23DSimMIM method) (vision_architectures.nets.swinv2_3d.SwinV23DStage method) (vision_architectures.nets.swinv2_3d.SwinV23DVAEMIM method) (vision_architectures.nets.symswin_3d.SymSwin3DBlock method) (vision_architectures.nets.symswin_3d.SymSwin3DEmbeddings method) (vision_architectures.nets.symswin_3d.SymSwin3DEncoder method) (vision_architectures.nets.symswin_3d.SymSwin3DLayer method) (vision_architectures.nets.symswin_3d.SymSwin3DLayerMLP method) (vision_architectures.nets.symswin_3d.SymSwin3DMHSA method) (vision_architectures.nets.symswin_3d.SymSwin3DModel method) (vision_architectures.nets.symswin_3d.SymSwin3DPatchEmbeddings method) (vision_architectures.nets.symswin_3d.SymSwin3DPatchMerging method) (vision_architectures.nets.symswin_3d.SymSwin3DReconstructionDecoder method) (vision_architectures.nets.symswin_3d.SymSwin3DSimMIM method) (vision_architectures.nets.symswin_3d.SymSwin3DStage method) (vision_architectures.nets.symswin_3d.SymSwin3DVAEMIM method) (vision_architectures.nets.unetr_3d_decoder.UNetR3DBlock method) (vision_architectures.nets.unetr_3d_decoder.UNetR3DConvBlock method) (vision_architectures.nets.unetr_3d_decoder.UNetR3DDecoder method) (vision_architectures.nets.unetr_3d_decoder.UNetR3DDeConvBlock method) (vision_architectures.nets.upernet_2d.UPerNet2D method) (vision_architectures.nets.upernet_2d.UPerNet2DFusion method) (vision_architectures.nets.upernet_3d.UPerNet3D method) (vision_architectures.nets.upernet_3d.UPerNet3DFusion method) (vision_architectures.nets.vit_3d.ViT3DDecoder method) (vision_architectures.nets.vit_3d.ViT3DEncoder method) (vision_architectures.nets.vit_3d.ViT3DMIMDecoder method) (vision_architectures.nets.vit_3d.ViT3DModel method) (vision_architectures.nets.vit_3d.ViT3DSimMIM method) (vision_architectures.utils.normalizations.DyT method) (vision_architectures.utils.normalizations.LayerNorm2D method) (vision_architectures.utils.normalizations.LayerNorm3D method) (vision_architectures.utils.pipeline_parallelism.PipelineModule method) (vision_architectures.utils.residuals.Residual method) (vision_architectures.utils.residuals.StochasticDepthResidual method) (vision_architectures.utils.timesteps.TimestepSampler method) FPN2D (class in vision_architectures.nets.fpn_2d) FPN2DBlock (class in vision_architectures.nets.fpn_2d) FPN3D (class in vision_architectures.nets.fpn_3d) FPN3DBlock (class in vision_architectures.nets.fpn_3d) fuse_features() (vision_architectures.nets.upernet_2d.UPerNet2DFusion method) (vision_architectures.nets.upernet_3d.UPerNet3DFusion method) fused_shape (vision_architectures.nets.upernet_2d.UPerNet2DFusionConfig attribute) (vision_architectures.nets.upernet_3d.UPerNet3DFusionConfig attribute) fusion (vision_architectures.nets.upernet_2d.UPerNet2DConfig attribute) (vision_architectures.nets.upernet_3d.UPerNet3DConfig attribute) G GaussianLatentSpace (class in vision_architectures.layers.latent_space) get() (vision_architectures.schedulers.cyclic.CyclicAnnealingScheduler method) (vision_architectures.schedulers.cyclic.SineScheduler method) (vision_architectures.schedulers.sigmoid.SigmoidScheduler method) (vision_architectures.utils.custom_base_model.CustomBaseModel method) get_3d_position_embeddings() (in module vision_architectures.nets.symswin_3d) get_absolute_position_embeddings_1d() (in module vision_architectures.layers.embeddings) get_absolute_position_embeddings_3d() (in module vision_architectures.layers.embeddings) get_act_layer() (in module vision_architectures.utils.activations) get_all_timestep_embeddings_1d() (in module vision_architectures.layers.embeddings) get_alpha_bars() (vision_architectures.schedulers.noise.NoiseScheduler method) get_alphas() (vision_architectures.schedulers.noise.NoiseScheduler method) get_beta() (vision_architectures.nets.swinv2_3d.SwinV23DVAEMIM method) (vision_architectures.nets.symswin_3d.SymSwin3DVAEMIM method) get_betas() (vision_architectures.schedulers.noise.NoiseScheduler method) get_coords_grid() (in module vision_architectures.layers.embeddings) (in module vision_architectures.nets.symswin_3d) get_data() (vision_architectures.image_readers.safetensors_reader.SafetensorsReader method) get_device() (in module vision_architectures.utils.pipeline_parallelism) get_edge_context() (vision_architectures.blocks.cnn.TensorSplittingConv method) get_expanded_shape() (vision_architectures.utils.splitter_merger.Splitter method) get_in_patch_size() (vision_architectures.nets.swin_3d.Swin3DStageConfig method) (vision_architectures.nets.swinv2_3d.SwinV23DStageConfig method) get_input_shape() (vision_architectures.blocks.cnn.TensorSplittingConv method) get_lr() (vision_architectures.schedulers.cyclic.CyclicAnnealingLR method) (vision_architectures.schedulers.cyclic.SineLR method) (vision_architectures.schedulers.sigmoid.SigmoidLR method) get_norm_layer() (in module vision_architectures.utils.normalizations) get_num_splits() (vision_architectures.utils.splitter_merger.Splitter method) get_optimized_num_splits() (vision_architectures.blocks.cnn.TensorSplittingConv method) get_out_patch_size() (vision_architectures.nets.swin_3d.Swin3DStageConfig method) (vision_architectures.nets.swinv2_3d.SwinV23DStageConfig method) get_positions() (vision_architectures.utils.splitter_merger.Splitter method) get_receptive_field() (vision_architectures.blocks.cnn.TensorSplittingConv method) get_relative_position_embeddings_table() (vision_architectures.layers.embeddings.RelativePositionEmbeddings3DMetaNetwork method) get_signal_to_noise_ratio() (vision_architectures.schedulers.noise.NoiseScheduler method) get_sinusoidal_embeddings_1d() (in module vision_architectures.layers.embeddings) get_sinusoidal_embeddings_3d() (in module vision_architectures.layers.embeddings) get_snr() (vision_architectures.schedulers.noise.NoiseScheduler method) get_specific_sinusoidal_embeddings_1d() (in module vision_architectures.layers.embeddings) get_split_size() (vision_architectures.blocks.cnn.TensorSplittingConv method) get_split_stride() (vision_architectures.blocks.cnn.TensorSplittingConv method) get_sqrt_alpha_bars() (vision_architectures.schedulers.noise.NoiseScheduler method) get_sqrt_one_minus_alpha_bars() (vision_architectures.schedulers.noise.NoiseScheduler method) get_timestep_embeddings_1d() (in module vision_architectures.layers.embeddings) get_updated_crop_start() (in module vision_architectures.transforms.croppad) get_velocity() (vision_architectures.schedulers.noise.NoiseScheduler method) gqa_mqa_enabled (vision_architectures.layers.attention.Attention1DConfig property) grid_size (vision_architectures.layers.embeddings.AbsolutePositionEmbeddings3DConfig attribute) (vision_architectures.layers.embeddings.RelativePositionEmbeddings3DConfig attribute) H hidden_dim (vision_architectures.blocks.mbconv_3d.MBConv3DConfig property) I image_size (vision_architectures.nets.swin_3d.Swin3DConfig attribute) (vision_architectures.nets.swinv2_3d.SwinV23DConfig attribute) (vision_architectures.nets.vit_3d.ViT3DConfig attribute) in_channels (vision_architectures.blocks.cnn.CNNBlockConfig attribute) (vision_architectures.blocks.mbconv_3d.MBConv3DConfig attribute) (vision_architectures.blocks.se.SEBlock3DConfig attribute) (vision_architectures.layers.embeddings.PatchEmbeddings3DConfig attribute) (vision_architectures.nets.fpn_2d.FPN2DBlockConfig attribute) (vision_architectures.nets.fpn_3d.FPN3DBlockConfig attribute) (vision_architectures.nets.maxvit_3d.MaxViT3DStem0Config attribute) (vision_architectures.nets.perceiver_3d.Perceiver3DChannelMappingConfig attribute) (vision_architectures.nets.swin_3d.Swin3DConfig attribute) (vision_architectures.nets.swinv2_3d.SwinV23DConfig attribute) (vision_architectures.nets.unetr_3d_decoder.UNetR3DConfig attribute) (vision_architectures.nets.upernet_2d.UPerNet2DFusionConfig attribute) (vision_architectures.nets.upernet_3d.UPerNet3DFusionConfig attribute) (vision_architectures.nets.vit_3d.ViT3DConfig attribute) in_dim (vision_architectures.nets.swin_3d.Swin3DStageConfig attribute) (vision_architectures.nets.swinv2_3d.SwinV23DStageConfig attribute) (vision_architectures.nets.unetr_3d_decoder.UNetR3DStageConfig attribute) in_patch_size (vision_architectures.nets.unetr_3d_decoder.UNetR3DStageConfig attribute) init_low_var (vision_architectures.layers.latent_space.LatentEncoderConfig attribute) init_low_var() (vision_architectures.layers.latent_space.LatentEncoder method) interpolation_mode (vision_architectures.nets.fpn_2d.FPN2DBlockConfig attribute) (vision_architectures.nets.fpn_3d.FPN3DBlockConfig attribute) (vision_architectures.nets.upernet_2d.UPerNet2DFusionConfig attribute) (vision_architectures.nets.upernet_3d.UPerNet3DFusionConfig attribute) inverse_transform() (vision_architectures.transforms.resize.ResizeWithSpacing method) is_completed() (vision_architectures.schedulers.sigmoid.SigmoidScheduler method) is_deepest (vision_architectures.nets.fpn_2d.FPN2DBlockConfig attribute) (vision_architectures.nets.fpn_3d.FPN3DBlockConfig attribute) is_ready() (vision_architectures.schedulers.cyclic.CyclicAnnealingScheduler method) (vision_architectures.schedulers.cyclic.SineScheduler method) (vision_architectures.schedulers.sigmoid.SigmoidScheduler method) K kernel_size (vision_architectures.blocks.cnn.CNNBlockConfig attribute) (vision_architectures.blocks.cnn.MultiResCNNBlockConfig attribute) (vision_architectures.blocks.mbconv_3d.MBConv3DConfig attribute) (vision_architectures.blocks.se.SEBlock3DConfig attribute) (vision_architectures.layers.embeddings.PatchEmbeddings3DConfig attribute) (vision_architectures.nets.fpn_2d.FPN2DBlockConfig attribute) (vision_architectures.nets.fpn_3d.FPN3DBlockConfig attribute) (vision_architectures.nets.maxvit_3d.MaxViT3DStem0Config attribute) (vision_architectures.nets.upernet_2d.UPerNet2DFusionConfig attribute) (vision_architectures.nets.upernet_3d.UPerNet3DFusionConfig attribute) kernel_sizes (vision_architectures.blocks.cnn.MultiResCNNBlockConfig attribute) kl_divergence() (vision_architectures.layers.latent_space.GaussianLatentSpace static method) kl_divergence_loss_fn() (vision_architectures.nets.swinv2_3d.SwinV23DVAEMIM static method) (vision_architectures.nets.symswin_3d.SymSwin3DVAEMIM static method) L latent_dim (vision_architectures.layers.latent_space.LatentDecoderConfig property) (vision_architectures.layers.latent_space.LatentEncoderConfig property) latent_grid_size (vision_architectures.nets.perceiver_3d.Perceiver3DEncoderConfig property) (vision_architectures.nets.perceiver_3d.Perceiver3DEncoderEncodeConfig attribute) (vision_architectures.nets.perceiver_3d.Perceiver3DEncoderProcessConfig attribute) LatentDecoder (class in vision_architectures.layers.latent_space) LatentEncoder (class in vision_architectures.layers.latent_space) layer_norm_eps (vision_architectures.blocks.transformer.Attention1DWithMLPConfig attribute) (vision_architectures.blocks.transformer.Attention3DWithMLPConfig attribute) (vision_architectures.nets.vit_3d.ViT3DDecoderConfig attribute) (vision_architectures.nets.vit_3d.ViT3DEncoderConfig attribute) LayerNorm2D (class in vision_architectures.utils.normalizations) LayerNorm3D (class in vision_architectures.utils.normalizations) learnable (vision_architectures.layers.embeddings.AbsolutePositionEmbeddings1DConfig attribute) (vision_architectures.layers.embeddings.AbsolutePositionEmbeddings3DConfig attribute) learnable_absolute_position_embeddings (vision_architectures.nets.swin_3d.Swin3DConfig attribute) (vision_architectures.nets.swinv2_3d.SwinV23DConfig attribute) length (vision_architectures.layers.embeddings.AbsolutePositionEmbeddings1DConfig attribute) LinearNoiseScheduler (class in vision_architectures.schedulers.noise) logit_scale_learnable (vision_architectures.layers.attention.Attention1DConfig attribute) (vision_architectures.nets.swin_3d.Swin3DStageConfig attribute) (vision_architectures.nets.swinv2_3d.SwinV23DStageConfig attribute) loss_fn() (vision_architectures.nets.swinv2_3d.SwinV23DSimMIM static method) (vision_architectures.nets.symswin_3d.SymSwin3DSimMIM static method) (vision_architectures.nets.unetr_3d_decoder.UNetR3DDecoder static method) (vision_architectures.nets.vit_3d.ViT3DSimMIM static method) M make_channels_first() (in module vision_architectures.utils.rearrange) make_channels_last() (in module vision_architectures.utils.rearrange) mask_image() (vision_architectures.nets.swinv2_3d.SwinV23DMIM method) (vision_architectures.nets.symswin_3d.SymSwin3DMIM method) (vision_architectures.nets.vit_3d.ViT3DMIM method) mask_ratio (vision_architectures.nets.vit_3d.ViT3DConfig attribute) max_attention_batch_size (vision_architectures.layers.attention.Attention1DConfig attribute) MaxViT3DBlock (class in vision_architectures.nets.maxvit_3d) MaxViT3DBlockAttention (class in vision_architectures.nets.maxvit_3d) MaxViT3DEncoder (class in vision_architectures.nets.maxvit_3d) MaxViT3DGridAttention (class in vision_architectures.nets.maxvit_3d) MaxViT3DStem (class in vision_architectures.nets.maxvit_3d) MaxViT3DStem0 (class in vision_architectures.nets.maxvit_3d) MBConv3D (class in vision_architectures.blocks.mbconv_3d) merge_method (vision_architectures.nets.fpn_2d.FPN2DBlockConfig attribute) (vision_architectures.nets.fpn_3d.FPN3DBlockConfig attribute) merge_window_size (vision_architectures.nets.swin_3d.Swin3DPatchMergingConfig attribute) (vision_architectures.nets.swinv2_3d.SwinV23DPatchMergingConfig attribute) Merger (class in vision_architectures.utils.splitter_merger) mim (vision_architectures.nets.swin_3d.Swin3DMIMConfig attribute) (vision_architectures.nets.swinv2_3d.Swin3DMIMConfig attribute) mlp_drop_prob (vision_architectures.blocks.transformer.Attention1DMLPConfig attribute) (vision_architectures.nets.vit_3d.ViT3DDecoderConfig attribute) (vision_architectures.nets.vit_3d.ViT3DEncoderConfig attribute) mlp_ratio (vision_architectures.blocks.transformer.Attention1DMLPConfig attribute) (vision_architectures.nets.vit_3d.ViT3DDecoderConfig attribute) (vision_architectures.nets.vit_3d.ViT3DEncoderConfig attribute) model_validate() (vision_architectures.utils.custom_base_model.CustomBaseModel class method) modify_dims (vision_architectures.nets.maxvit_3d.MaxViT3DBlockConfig attribute) module vision_architectures.blocks.cnn vision_architectures.blocks.heads_3d vision_architectures.blocks.mbconv_3d vision_architectures.blocks.se vision_architectures.blocks.transformer vision_architectures.docstrings vision_architectures.image_readers.safetensors_reader vision_architectures.layers.attention vision_architectures.layers.codebook vision_architectures.layers.embeddings vision_architectures.layers.latent_space vision_architectures.layers.scale vision_architectures.nets.fpn_2d vision_architectures.nets.fpn_3d vision_architectures.nets.maxvit_3d vision_architectures.nets.perceiver_3d vision_architectures.nets.swin_3d vision_architectures.nets.swinv2_3d vision_architectures.nets.symswin_3d vision_architectures.nets.unetr_3d_decoder vision_architectures.nets.upernet_2d vision_architectures.nets.upernet_3d vision_architectures.nets.vit_3d vision_architectures.schedulers.cyclic vision_architectures.schedulers.lrs vision_architectures.schedulers.noise vision_architectures.schedulers.sigmoid vision_architectures.transforms.clipping vision_architectures.transforms.croppad vision_architectures.transforms.resize vision_architectures.transforms.spatial vision_architectures.utils.activation_checkpointing vision_architectures.utils.activations vision_architectures.utils.clamping vision_architectures.utils.custom_base_model vision_architectures.utils.normalizations vision_architectures.utils.pipeline_parallelism vision_architectures.utils.rearrange vision_architectures.utils.residuals vision_architectures.utils.splitter_merger vision_architectures.utils.timesteps move_to_device() (in module vision_architectures.utils.pipeline_parallelism) MultiResCNNBlock2D (class in vision_architectures.blocks.cnn) MultiResCNNBlock3D (class in vision_architectures.blocks.cnn) N NoiseScheduler (class in vision_architectures.schedulers.noise) norm_layer (vision_architectures.layers.embeddings.PatchEmbeddings3DConfig attribute) norm_location (vision_architectures.blocks.transformer.Attention1DWithMLPConfig attribute) (vision_architectures.blocks.transformer.Attention3DWithMLPConfig attribute) (vision_architectures.nets.vit_3d.ViT3DDecoderConfig attribute) (vision_architectures.nets.vit_3d.ViT3DEncoderConfig attribute) normalization (vision_architectures.blocks.cnn.CNNBlockConfig attribute) (vision_architectures.blocks.mbconv_3d.MBConv3DConfig attribute) (vision_architectures.blocks.se.SEBlock3DConfig attribute) (vision_architectures.nets.fpn_2d.FPN2DBlockConfig attribute) (vision_architectures.nets.upernet_2d.UPerNet2DFusionConfig attribute) normalization_kwargs (vision_architectures.blocks.cnn.CNNBlockConfig attribute) normalization_post_args (vision_architectures.blocks.cnn.CNNBlockConfig attribute) normalization_pre_args (vision_architectures.blocks.cnn.CNNBlockConfig attribute) NOT_READY_ERROR_MSG (vision_architectures.schedulers.cyclic.CyclicAnnealingScheduler attribute) num_class_tokens (vision_architectures.nets.vit_3d.ViT3DConfig attribute) num_features (vision_architectures.nets.upernet_2d.UPerNet2DFusionConfig attribute) (vision_architectures.nets.upernet_3d.UPerNet3DFusionConfig attribute) num_heads (vision_architectures.layers.attention.Attention1DConfig attribute) (vision_architectures.layers.embeddings.RelativePositionEmbeddings3DConfig attribute) (vision_architectures.nets.vit_3d.ViT3DDecoderConfig attribute) (vision_architectures.nets.vit_3d.ViT3DEncoderConfig attribute) num_kv_heads (vision_architectures.layers.attention.Attention1DConfig property) num_layers (vision_architectures.nets.perceiver_3d.Perceiver3DDecoderConfig attribute) (vision_architectures.nets.perceiver_3d.Perceiver3DEncoderEncodeConfig attribute) (vision_architectures.nets.perceiver_3d.Perceiver3DEncoderProcessConfig attribute) num_objects (vision_architectures.nets.upernet_2d.UPerNet2DConfig attribute) (vision_architectures.nets.upernet_3d.UPerNet3DConfig attribute) num_outputs (vision_architectures.nets.unetr_3d_decoder.UNetR3DDecoderConfig attribute) num_patches (vision_architectures.layers.embeddings.AbsolutePositionEmbeddings3DConfig property) (vision_architectures.layers.embeddings.RelativePositionEmbeddings3DConfig property) num_q_heads (vision_architectures.layers.attention.Attention1DConfig property) num_vectors (vision_architectures.layers.codebook.CodebookConfig attribute) O out_channels (vision_architectures.blocks.cnn.CNNBlockConfig attribute) (vision_architectures.blocks.mbconv_3d.MBConv3DConfig attribute) (vision_architectures.blocks.se.SEBlock3DConfig attribute) (vision_architectures.layers.embeddings.PatchEmbeddings3DConfig attribute) (vision_architectures.nets.fpn_2d.FPN2DBlockConfig attribute) (vision_architectures.nets.fpn_3d.FPN3DBlockConfig attribute) (vision_architectures.nets.maxvit_3d.MaxViT3DStem0Config attribute) (vision_architectures.nets.perceiver_3d.Perceiver3DChannelMappingConfig attribute) (vision_architectures.nets.perceiver_3d.Perceiver3DDecoderConfig attribute) (vision_architectures.nets.upernet_2d.UPerNet2DFusionConfig attribute) (vision_architectures.nets.upernet_3d.UPerNet3DFusionConfig attribute) out_dim (vision_architectures.blocks.mbconv_3d.MBConv3DConfig attribute) (vision_architectures.nets.swin_3d.Swin3DStageConfig attribute) (vision_architectures.nets.swinv2_3d.SwinV23DStageConfig attribute) (vision_architectures.nets.unetr_3d_decoder.UNetR3DStageConfig attribute) out_dim_ratio (vision_architectures.nets.maxvit_3d.MaxViT3DBlockConfig attribute) (vision_architectures.nets.swin_3d.Swin3DPatchMergingConfig attribute) (vision_architectures.nets.swin_3d.Swin3DPatchSplittingConfig attribute) (vision_architectures.nets.swinv2_3d.SwinV23DPatchMergingConfig attribute) (vision_architectures.nets.swinv2_3d.SwinV23DPatchSplittingConfig attribute) out_patch_size (vision_architectures.nets.unetr_3d_decoder.UNetR3DStageConfig attribute) P pad_input_for_context() (vision_architectures.blocks.cnn.TensorSplittingConv method) pad_input_for_divisibility() (vision_architectures.blocks.cnn.TensorSplittingConv method) padding (vision_architectures.blocks.cnn.CNNBlockConfig attribute) (vision_architectures.blocks.cnn.MultiResCNNBlockConfig attribute) paralellize_pipeline() (in module vision_architectures.utils.pipeline_parallelism) patch_merging (vision_architectures.nets.swin_3d.Swin3DStageConfig attribute) (vision_architectures.nets.swinv2_3d.SwinV23DStageConfig attribute) patch_size (vision_architectures.layers.embeddings.PatchEmbeddings3DConfig attribute) (vision_architectures.nets.swin_3d.Swin3DConfig attribute) (vision_architectures.nets.swinv2_3d.SwinV23DConfig attribute) (vision_architectures.nets.vit_3d.ViT3DConfig attribute) patch_splitting (vision_architectures.nets.swin_3d.Swin3DStageConfig attribute) (vision_architectures.nets.swinv2_3d.SwinV23DStageConfig attribute) PatchEmbeddings3D (class in vision_architectures.layers.embeddings) per_head_dim_qk (vision_architectures.layers.attention.Attention1DConfig property) Perceiver3DChannelMapping (class in vision_architectures.nets.perceiver_3d) Perceiver3DDecoder (class in vision_architectures.nets.perceiver_3d) Perceiver3DEncoder (class in vision_architectures.nets.perceiver_3d) Perceiver3DEncoderEncode (class in vision_architectures.nets.perceiver_3d) Perceiver3DEncoderProcess (class in vision_architectures.nets.perceiver_3d) Phase (class in vision_architectures.schedulers.cyclic) PipelineModule (class in vision_architectures.utils.pipeline_parallelism) PixelShuffleDownsample3D (class in vision_architectures.layers.scale) PixelShuffleUpsample3D (class in vision_architectures.layers.scale) populate() (vision_architectures.nets.swin_3d.Swin3DConfig method) (vision_architectures.nets.swinv2_3d.SwinV23DDecoderConfig method) populate_and_validate_config() (in module vision_architectures.nets.symswin_3d) populate_docstring() (in module vision_architectures.docstrings) process (vision_architectures.nets.perceiver_3d.Perceiver3DEncoderConfig attribute) proj_drop_prob (vision_architectures.layers.attention.Attention1DConfig attribute) (vision_architectures.nets.vit_3d.ViT3DDecoderConfig attribute) (vision_architectures.nets.vit_3d.ViT3DEncoderConfig attribute) Q quantize() (vision_architectures.layers.codebook.Codebook method) R r (vision_architectures.blocks.se.SEBlock3DConfig attribute) raise_large_stride_error (vision_architectures.utils.splitter_merger.SplitterConfig attribute) RandFlipWithCropTrackingd (class in vision_architectures.transforms.spatial) RandSpatialCropSamplesWithCropTracking (class in vision_architectures.transforms.croppad) RandSpatialCropSamplesWithCropTrackingd (class in vision_architectures.transforms.croppad) ratio_q_to_kv_heads (vision_architectures.layers.attention.Attention1DConfig attribute) read() (vision_architectures.image_readers.safetensors_reader.SafetensorsReader method) rearrange_channels() (in module vision_architectures.utils.rearrange) reconstruction_loss_fn() (vision_architectures.nets.swinv2_3d.SwinV23DVAEMIM static method) (vision_architectures.nets.symswin_3d.SymSwin3DVAEMIM static method) RelativePositionEmbeddings3D (class in vision_architectures.layers.embeddings) RelativePositionEmbeddings3DMetaNetwork (class in vision_architectures.layers.embeddings) remove_noise() (vision_architectures.schedulers.noise.NoiseScheduler method) remove_stochastic_depth_dropout() (in module vision_architectures.utils.residuals) remove_tsp_from_module() (in module vision_architectures.blocks.cnn) reparameterize() (vision_architectures.nets.swinv2_3d.SwinV23DVAEMIM method) (vision_architectures.nets.symswin_3d.SymSwin3DVAEMIM method) Residual (class in vision_architectures.utils.residuals) ResizedWithCropTrackingd (class in vision_architectures.transforms.spatial) ResizeWithSpacing (class in vision_architectures.transforms.resize) revive_dead_vectors() (vision_architectures.layers.codebook.Codebook method) revive_dead_vectors_after_n_steps (vision_architectures.layers.codebook.CodebookConfig attribute) S SafetensorsReader (class in vision_architectures.image_readers.safetensors_reader) sample() (vision_architectures.layers.latent_space.GaussianLatentSpace method) scale_factor (vision_architectures.layers.scale.PixelShuffleScaleConfig attribute) scale_filter_ratios() (vision_architectures.blocks.cnn.MultiResCNNBlockConfig class method) se_reduction_ratio (vision_architectures.blocks.mbconv_3d.MBConv3DConfig attribute) SEBlock3D (class in vision_architectures.blocks.se) SegmentationHead3D (class in vision_architectures.blocks.heads_3d) sequence (vision_architectures.blocks.cnn.CNNBlockConfig attribute) set_next_phase() (vision_architectures.schedulers.cyclic.CyclicAnnealingScheduler method) set_num_annealing_steps() (vision_architectures.schedulers.cyclic.CyclicAnnealingScheduler method) set_num_steps() (vision_architectures.schedulers.sigmoid.SigmoidScheduler method) set_wavelength() (vision_architectures.schedulers.cyclic.SineScheduler method) setdefault() (vision_architectures.utils.custom_base_model.CustomBaseModel method) SigmoidLR (class in vision_architectures.schedulers.sigmoid) SigmoidNoiseScheduler (class in vision_architectures.schedulers.noise) SigmoidScheduler (class in vision_architectures.schedulers.sigmoid) SineLR (class in vision_architectures.schedulers.cyclic) SineScheduler (class in vision_architectures.schedulers.cyclic) skip_conn_dim (vision_architectures.nets.fpn_2d.FPN2DBlockConfig attribute) (vision_architectures.nets.fpn_3d.FPN3DBlockConfig attribute) soft_dice_loss_fn() (vision_architectures.nets.unetr_3d_decoder.UNetR3DDecoder static method) spatial_compression_ratio (vision_architectures.nets.swinv2_3d.SwinV23DStageConfig property) split() (vision_architectures.utils.splitter_merger.Splitter method) split_dims (vision_architectures.utils.splitter_merger.SplitterConfig attribute) split_size (vision_architectures.utils.splitter_merger.SplitterConfig attribute) Splitter (class in vision_architectures.utils.splitter_merger) SquareRootNoiseScheduler (class in vision_architectures.schedulers.noise) stages (vision_architectures.nets.swin_3d.Swin3DConfig attribute) (vision_architectures.nets.swinv2_3d.SwinV23DDecoderConfig attribute) (vision_architectures.nets.unetr_3d_decoder.UNetR3DConfig attribute) stem0 (vision_architectures.nets.maxvit_3d.MaxViT3DEncoderConfig attribute) stems (vision_architectures.nets.maxvit_3d.MaxViT3DEncoderConfig attribute) step() (vision_architectures.schedulers.cyclic.CyclicAnnealingLR method) (vision_architectures.schedulers.cyclic.CyclicAnnealingScheduler method) (vision_architectures.schedulers.cyclic.SineLR method) (vision_architectures.schedulers.cyclic.SineScheduler method) (vision_architectures.schedulers.sigmoid.SigmoidLR method) (vision_architectures.schedulers.sigmoid.SigmoidScheduler method) StochasticDepthResidual (class in vision_architectures.utils.residuals) stride (vision_architectures.blocks.cnn.CNNBlockConfig attribute) (vision_architectures.utils.splitter_merger.SplitterConfig attribute) Swin3DBlock (class in vision_architectures.nets.swin_3d) Swin3DEncoder (class in vision_architectures.nets.swin_3d) Swin3DLayer (class in vision_architectures.nets.swin_3d) Swin3DMIM (class in vision_architectures.nets.swin_3d) Swin3DMIMDecoder (class in vision_architectures.nets.swin_3d) Swin3DModel (class in vision_architectures.nets.swin_3d) Swin3DPatchMerging (class in vision_architectures.nets.swin_3d) Swin3DStage (class in vision_architectures.nets.swin_3d) SwinV23DBlock (class in vision_architectures.nets.swinv2_3d) SwinV23DDecoder (class in vision_architectures.nets.swinv2_3d) SwinV23DEncoder (class in vision_architectures.nets.swinv2_3d) SwinV23DLayer (class in vision_architectures.nets.swinv2_3d) SwinV23DLayerLogitScale (class in vision_architectures.nets.swinv2_3d) SwinV23DMIM (class in vision_architectures.nets.swinv2_3d) SwinV23DModel (class in vision_architectures.nets.swinv2_3d) SwinV23DPatchMerging (class in vision_architectures.nets.swinv2_3d) SwinV23DPatchSplitting (class in vision_architectures.nets.swinv2_3d) SwinV23DReconstructionDecoder (class in vision_architectures.nets.swinv2_3d) SwinV23DSimMIM (class in vision_architectures.nets.swinv2_3d) SwinV23DStage (class in vision_architectures.nets.swinv2_3d) SwinV23DVAEMIM (class in vision_architectures.nets.swinv2_3d) symmetric_tanh_clamp() (in module vision_architectures.utils.clamping) symmetry_attention_rearrange_backward() (in module vision_architectures.nets.symswin_3d) symmetry_attention_rearrange_forward() (in module vision_architectures.nets.symswin_3d) SymSwin3DBlock (class in vision_architectures.nets.symswin_3d) SymSwin3DEmbeddings (class in vision_architectures.nets.symswin_3d) SymSwin3DEncoder (class in vision_architectures.nets.symswin_3d) SymSwin3DLayer (class in vision_architectures.nets.symswin_3d) SymSwin3DLayerMLP (class in vision_architectures.nets.symswin_3d) SymSwin3DMHSA (class in vision_architectures.nets.symswin_3d) SymSwin3DMIM (class in vision_architectures.nets.symswin_3d) SymSwin3DModel (class in vision_architectures.nets.symswin_3d) SymSwin3DPatchEmbeddings (class in vision_architectures.nets.symswin_3d) SymSwin3DPatchMerging (class in vision_architectures.nets.symswin_3d) SymSwin3DReconstructionDecoder (class in vision_architectures.nets.symswin_3d) SymSwin3DSimMIM (class in vision_architectures.nets.symswin_3d) SymSwin3DStage (class in vision_architectures.nets.symswin_3d) SymSwin3DVAEMIM (class in vision_architectures.nets.symswin_3d) T TensorSplittingConv (class in vision_architectures.blocks.cnn) TimestepSampler (class in vision_architectures.utils.timesteps) TOP (vision_architectures.schedulers.cyclic.Phase attribute) TransformerDecoderBlock1D (class in vision_architectures.blocks.transformer) TransformerDecoderBlock3D (class in vision_architectures.blocks.transformer) TransformerEncoderBlock1D (class in vision_architectures.blocks.transformer) TransformerEncoderBlock3D (class in vision_architectures.blocks.transformer) transposed (vision_architectures.blocks.cnn.CNNBlockConfig attribute) U UNetR3DBlock (class in vision_architectures.nets.unetr_3d_decoder) UNetR3DConvBlock (class in vision_architectures.nets.unetr_3d_decoder) UNetR3DDecoder (class in vision_architectures.nets.unetr_3d_decoder) UNetR3DDeConvBlock (class in vision_architectures.nets.unetr_3d_decoder) unfold_with_roll_3d() (in module vision_architectures.nets.perceiver_3d) UP (vision_architectures.schedulers.cyclic.Phase attribute) UPerNet2D (class in vision_architectures.nets.upernet_2d) UPerNet2DFusion (class in vision_architectures.nets.upernet_2d) UPerNet3D (class in vision_architectures.nets.upernet_3d) UPerNet3DFusion (class in vision_architectures.nets.upernet_3d) use_absolute_position_embeddings (vision_architectures.nets.perceiver_3d.Perceiver3DDecoderConfig attribute) (vision_architectures.nets.swin_3d.Swin3DConfig attribute) (vision_architectures.nets.swinv2_3d.SwinV23DConfig attribute) use_relative_position_bias (vision_architectures.nets.swin_3d.Swin3DStageConfig attribute) (vision_architectures.nets.swinv2_3d.SwinV23DStageConfig attribute) use_relative_position_embeddings (vision_architectures.nets.perceiver_3d.Perceiver3DEncoderProcessConfig attribute) V validate() (vision_architectures.blocks.cnn.CNNBlockConfig method) (vision_architectures.blocks.cnn.MultiResCNNBlockConfig method) (vision_architectures.blocks.mbconv_3d.MBConv3DConfig method) (vision_architectures.layers.attention.Attention1DConfig method) (vision_architectures.layers.embeddings.AbsolutePositionEmbeddings1DConfig method) (vision_architectures.layers.embeddings.AbsolutePositionEmbeddings3DConfig method) (vision_architectures.layers.embeddings.RelativePositionEmbeddings3DConfig method) (vision_architectures.nets.fpn_2d.FPN2DConfig method) (vision_architectures.nets.fpn_3d.FPN3DConfig method) (vision_architectures.nets.maxvit_3d.MaxViT3DEncoderConfig method) (vision_architectures.nets.perceiver_3d.Perceiver3DConfig method) (vision_architectures.nets.perceiver_3d.Perceiver3DEncoderConfig method) (vision_architectures.nets.perceiver_3d.Perceiver3DEncoderProcessConfig method) (vision_architectures.nets.swin_3d.Swin3DConfig method) (vision_architectures.nets.swinv2_3d.SwinV23DConfig method) (vision_architectures.nets.swinv2_3d.SwinV23DDecoderConfig method) (vision_architectures.nets.unetr_3d_decoder.UNetR3DConfig method) (vision_architectures.nets.upernet_2d.UPerNet2DConfig method) (vision_architectures.nets.upernet_3d.UPerNet3DConfig method) (vision_architectures.utils.custom_base_model.CustomBaseModel method) (vision_architectures.utils.splitter_merger.SplitterConfig method) validate_before() (vision_architectures.blocks.mbconv_3d.MBConv3DConfig class method) (vision_architectures.layers.embeddings.AbsolutePositionEmbeddings3DConfig class method) (vision_architectures.layers.embeddings.PatchEmbeddings3DConfig class method) (vision_architectures.layers.embeddings.RelativePositionEmbeddings3DConfig class method) (vision_architectures.layers.latent_space.LatentDecoderConfig class method) (vision_architectures.layers.latent_space.LatentEncoderConfig class method) (vision_architectures.nets.fpn_2d.FPN2DConfig class method) (vision_architectures.nets.fpn_3d.FPN3DConfig class method) (vision_architectures.nets.perceiver_3d.Perceiver3DConfig class method) (vision_architectures.nets.perceiver_3d.Perceiver3DEncoderConfig class method) (vision_architectures.nets.swin_3d.Swin3DPatchMergingConfig class method) (vision_architectures.nets.swin_3d.Swin3DPatchSplittingConfig class method) (vision_architectures.nets.swinv2_3d.SwinV23DPatchMergingConfig class method) (vision_architectures.nets.swinv2_3d.SwinV23DPatchSplittingConfig class method) (vision_architectures.nets.upernet_2d.UPerNet2DConfig class method) (vision_architectures.nets.upernet_3d.UPerNet3DConfig class method) (vision_architectures.utils.custom_base_model.CustomBaseModel class method) verify_suffix() (vision_architectures.image_readers.safetensors_reader.SafetensorsReader method) vision_architectures.blocks.cnn module vision_architectures.blocks.heads_3d module vision_architectures.blocks.mbconv_3d module vision_architectures.blocks.se module vision_architectures.blocks.transformer module vision_architectures.docstrings module vision_architectures.image_readers.safetensors_reader module vision_architectures.layers.attention module vision_architectures.layers.codebook module vision_architectures.layers.embeddings module vision_architectures.layers.latent_space module vision_architectures.layers.scale module vision_architectures.nets.fpn_2d module vision_architectures.nets.fpn_3d module vision_architectures.nets.maxvit_3d module vision_architectures.nets.perceiver_3d module vision_architectures.nets.swin_3d module vision_architectures.nets.swinv2_3d module vision_architectures.nets.symswin_3d module vision_architectures.nets.unetr_3d_decoder module vision_architectures.nets.upernet_2d module vision_architectures.nets.upernet_3d module vision_architectures.nets.vit_3d module vision_architectures.schedulers.cyclic module vision_architectures.schedulers.lrs module vision_architectures.schedulers.noise module vision_architectures.schedulers.sigmoid module vision_architectures.transforms.clipping module vision_architectures.transforms.croppad module vision_architectures.transforms.resize module vision_architectures.transforms.spatial module vision_architectures.utils.activation_checkpointing module vision_architectures.utils.activations module vision_architectures.utils.clamping module vision_architectures.utils.custom_base_model module vision_architectures.utils.normalizations module vision_architectures.utils.pipeline_parallelism module vision_architectures.utils.rearrange module vision_architectures.utils.residuals module vision_architectures.utils.splitter_merger module vision_architectures.utils.timesteps module ViT3DDecoder (class in vision_architectures.nets.vit_3d) ViT3DEncoder (class in vision_architectures.nets.vit_3d) ViT3DMIM (class in vision_architectures.nets.vit_3d) ViT3DMIMDecoder (class in vision_architectures.nets.vit_3d) ViT3DModel (class in vision_architectures.nets.vit_3d) ViT3DSimMIM (class in vision_architectures.nets.vit_3d) W window_size (vision_architectures.nets.maxvit_3d.MaxViT3DBlockConfig attribute) (vision_architectures.nets.swin_3d.Swin3DStageConfig attribute) (vision_architectures.nets.swinv2_3d.SwinV23DStageConfig attribute)