Index

_

__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.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.embeddings.RotaryPositionEmbeddings1D method) (vision_architectures.layers.embeddings.RotaryPositionEmbeddings3D method) (vision_architectures.layers.latent_space.GaussianLatentSpace method) (vision_architectures.layers.latent_space.LatentDecoder3D method) (vision_architectures.layers.latent_space.LatentEncoder3D method) (vision_architectures.layers.scale.PixelShuffleDownsample3D method) (vision_architectures.layers.scale.PixelShuffleUpsample3D method) (vision_architectures.losses.class_balanced_cross_entropy_loss.ClassBalancedCrossEntropyLoss method) (vision_architectures.metrics.detection.AveragePrecision method) (vision_architectures.metrics.detection.AverageRecall method) (vision_architectures.nets.cait_3d.CaiT1D method) (vision_architectures.nets.cait_3d.CaiTAttentionWithMLP method) (vision_architectures.nets.cait_3d.CaiTStage1 method) (vision_architectures.nets.cait_3d.CaiTStage2 method) (vision_architectures.nets.detr_3d.DETR3D method) (vision_architectures.nets.detr_3d.DETR3DDecoder method) (vision_architectures.nets.detr_3d.DETR3DDecoderLayer method) (vision_architectures.nets.detr_3d.DETR3DEncoder method) (vision_architectures.nets.detr_3d.DETR3DEncoderLayer method) (vision_architectures.nets.detr_3d.DETRBBoxMLP 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.Swin3DDecoder method) (vision_architectures.nets.swin_3d.Swin3DEncoder method) (vision_architectures.nets.swin_3d.Swin3DEncoderDecoderBase method) (vision_architectures.nets.swin_3d.Swin3DEncoderWithPatchEmbeddings method) (vision_architectures.nets.swin_3d.Swin3DLayer method) (vision_architectures.nets.swin_3d.Swin3DPatchMerging method) (vision_architectures.nets.swin_3d.Swin3DPatchSplitting 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.SwinV23DEncoderDecoderBase method) (vision_architectures.nets.swinv2_3d.SwinV23DEncoderWithPatchEmbeddings method) (vision_architectures.nets.swinv2_3d.SwinV23DLayer method) (vision_architectures.nets.swinv2_3d.SwinV23DLayerLogitScale method) (vision_architectures.nets.swinv2_3d.SwinV23DPatchMerging method) (vision_architectures.nets.swinv2_3d.SwinV23DPatchSplitting method) (vision_architectures.nets.swinv2_3d.SwinV23DStage method) (vision_architectures.nets.unetr_3d_decoder.UNetR3DDecoder 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.ViT3DDecoderWithPatchEmbeddings method) (vision_architectures.nets.vit_3d.ViT3DEncoder method) (vision_architectures.nets.vit_3d.ViT3DEncoderWithPatchEmbeddings 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.ema_network.EMANetwork method) (vision_architectures.utils.normalizations.DyT 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

absolute_position_embeddings_config (vision_architectures.nets.vit_3d.ViT3DDecoderWithPatchEmbeddingsConfig attribute) (vision_architectures.nets.vit_3d.ViT3DEncoderWithPatchEmbeddingsConfig attribute) 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) (vision_architectures.nets.detr_3d.DETRBBoxMLPConfig 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) apply_rope() (vision_architectures.layers.embeddings.RotaryPositionEmbeddings1D method) (vision_architectures.layers.embeddings.RotaryPositionEmbeddings3D method) 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) AveragePrecision (class in vision_architectures.metrics.detection) AverageRecall (class in vision_architectures.metrics.detection)

B

base (vision_architectures.layers.embeddings.RotaryPositionEmbeddings1DConfig attribute) bbox_giou() (vision_architectures.nets.detr_3d.DETR3D static method) bbox_iou() (vision_architectures.nets.detr_3d.DETR3D static method) bbox_size_activation (vision_architectures.nets.detr_3d.DETRBBoxMLPConfig attribute)

bipartite_matching_loss() (vision_architectures.nets.detr_3d.DETR3D method) blocks (vision_architectures.nets.fpn_2d.FPN2DConfig attribute) (vision_architectures.nets.fpn_3d.FPN3DConfig attribute) BOTTOM (vision_architectures.schedulers.cyclic.Phase attribute)

C

CaiT1D (class in vision_architectures.nets.cait_3d) CaiT3D (class in vision_architectures.nets.cait_3d) CaiTAttentionWithMLP (class in vision_architectures.nets.cait_3d) CaiTStage1 (class in vision_architectures.nets.cait_3d) CaiTStage2 (class in vision_architectures.nets.cait_3d) calculate_losses() (vision_architectures.layers.codebook.Codebook method) calculate_perplexity() (vision_architectures.layers.codebook.Codebook method) ceil_softplus_clamp() (in module vision_architectures.utils.clamping) ClassBalancedCrossEntropyLoss (class in vision_architectures.losses.class_balanced_cross_entropy_loss) classification_cost_fn (vision_architectures.nets.detr_3d.DETR3DConfig attribute) classification_loss_fn (vision_architectures.nets.detr_3d.DETR3DConfig attribute) 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) compute() (vision_architectures.metrics.detection.MeanAveragePrecision method) (vision_architectures.metrics.detection.MeanAverageRecall method) 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) DETR3D (class in vision_architectures.nets.detr_3d) DETR3DDecoder (class in vision_architectures.nets.detr_3d) DETR3DDecoderLayer (class in vision_architectures.nets.detr_3d) DETR3DEncoder (class in vision_architectures.nets.detr_3d) DETR3DEncoderLayer (class in vision_architectures.nets.detr_3d) DETRBBoxMLP (class in vision_architectures.nets.detr_3d) dim (vision_architectures.blocks.mbconv_3d.MBConv3DConfig attribute) (vision_architectures.blocks.se.SEBlock3DConfig attribute) (vision_architectures.blocks.transformer.Attention1DMLPConfig 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.embeddings.RotaryPositionEmbeddings1DConfig attribute) (vision_architectures.layers.latent_space.LatentDecoderConfig property) (vision_architectures.layers.latent_space.LatentEncoderConfig property) (vision_architectures.nets.detr_3d.DETRBBoxMLPConfig attribute) (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.Swin3DBlockConfig attribute) (vision_architectures.nets.upernet_2d.UPerNet2DFusionConfig attribute) (vision_architectures.nets.upernet_3d.UPerNet3DFusionConfig 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) DyT (class in vision_architectures.utils.normalizations)

E

ema_decay (vision_architectures.layers.codebook.CodebookConfig attribute) (vision_architectures.losses.class_balanced_cross_entropy_loss.ClassBalancedCrossEntropyLossConfig attribute) EMANetwork (class in vision_architectures.utils.ema_network) 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.layers.embeddings.RotaryPositionEmbeddings1D method) (vision_architectures.layers.embeddings.RotaryPositionEmbeddings3D method) (vision_architectures.utils.activation_checkpointing.ActivationCheckpointing method) (vision_architectures.utils.normalizations.DyT 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) 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.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.embeddings.RotaryPositionEmbeddings1D method) (vision_architectures.layers.embeddings.RotaryPositionEmbeddings3D method) (vision_architectures.layers.latent_space.GaussianLatentSpace method) (vision_architectures.layers.latent_space.LatentDecoder3D method) (vision_architectures.layers.latent_space.LatentEncoder3D method) (vision_architectures.layers.scale.PixelShuffleDownsample3D method) (vision_architectures.layers.scale.PixelShuffleUpsample3D method) (vision_architectures.losses.class_balanced_cross_entropy_loss.ClassBalancedCrossEntropyLoss method) (vision_architectures.nets.cait_3d.CaiT1D method) (vision_architectures.nets.cait_3d.CaiTAttentionWithMLP method) (vision_architectures.nets.cait_3d.CaiTStage1 method) (vision_architectures.nets.cait_3d.CaiTStage2 method) (vision_architectures.nets.detr_3d.DETR3D method) (vision_architectures.nets.detr_3d.DETR3DDecoder method) (vision_architectures.nets.detr_3d.DETR3DDecoderLayer method) (vision_architectures.nets.detr_3d.DETR3DEncoder method) (vision_architectures.nets.detr_3d.DETR3DEncoderLayer method) (vision_architectures.nets.detr_3d.DETRBBoxMLP 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.Swin3DEncoderDecoderBase method) (vision_architectures.nets.swin_3d.Swin3DEncoderWithPatchEmbeddings method) (vision_architectures.nets.swin_3d.Swin3DLayer method) (vision_architectures.nets.swin_3d.Swin3DPatchMerging method) (vision_architectures.nets.swin_3d.Swin3DPatchSplitting method) (vision_architectures.nets.swin_3d.Swin3DStage method) (vision_architectures.nets.swinv2_3d.SwinV23DLayerLogitScale method) (vision_architectures.nets.unetr_3d_decoder.UNetR3DDecoder 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.ViT3DDecoderWithPatchEmbeddings method) (vision_architectures.nets.vit_3d.ViT3DEncoder method) (vision_architectures.nets.vit_3d.ViT3DEncoderWithPatchEmbeddings method) (vision_architectures.utils.ema_network.EMANetwork method) (vision_architectures.utils.normalizations.DyT method) (vision_architectures.utils.normalizations.LayerNorm2D method) (vision_architectures.utils.normalizations.LayerNorm3D 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_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_betas() (vision_architectures.schedulers.noise.NoiseScheduler method) get_class_prevalences() (vision_architectures.losses.class_balanced_cross_entropy_loss.ClassBalancedCrossEntropyLoss method) get_class_weights() (vision_architectures.losses.class_balanced_cross_entropy_loss.ClassBalancedCrossEntropyLoss method) get_coords_grid() (in module vision_architectures.layers.embeddings) get_data() (vision_architectures.image_readers.safetensors_reader.SafetensorsReader method) get_decay() (vision_architectures.utils.ema_network.EMANetwork 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_dim() (vision_architectures.nets.swin_3d.Swin3DBlockConfig method) get_in_patch_size() (vision_architectures.nets.swin_3d.Swin3DBlockConfig 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_dim() (vision_architectures.nets.swin_3d.Swin3DBlockConfig method) get_out_dim_ratios() (vision_architectures.nets.swin_3d.Swin3DEncoderDecoderConfig method) get_out_patch_size() (vision_architectures.nets.swin_3d.Swin3DBlockConfig 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_rope_rotation_coefficients_1d() (in module vision_architectures.layers.embeddings) get_rotation_coefficients() (vision_architectures.layers.embeddings.RotaryPositionEmbeddings1D static 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_as_ints() (vision_architectures.layers.embeddings.RotaryPositionEmbeddings3DConfig method) 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_tps_fps_fns() (in module vision_architectures.utils.bounding_boxes) 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)

hungarian_matching() (vision_architectures.nets.detr_3d.DETR3D method)

I

image_size (vision_architectures.nets.swin_3d.Swin3DEncoderWithPatchEmbeddingsConfig 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.Swin3DEncoderWithPatchEmbeddingsConfig attribute) (vision_architectures.nets.unetr_3d_decoder.UNetR3DConfig attribute) (vision_architectures.nets.upernet_2d.UPerNet2DFusionConfig attribute) (vision_architectures.nets.upernet_3d.UPerNet3DFusionConfig attribute) in_dim (vision_architectures.nets.swin_3d.Swin3DBlockConfig attribute) (vision_architectures.nets.swin_3d.Swin3DPatchMergingConfig attribute) (vision_architectures.nets.swin_3d.Swin3DPatchSplittingConfig attribute) (vision_architectures.nets.unetr_3d_decoder.UNetR3DStageConfig attribute)

in_dim_ratio (vision_architectures.nets.swin_3d.Swin3DPatchSplittingConfig property) 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.LatentEncoder3D 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_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)

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) LatentDecoder3D (class in vision_architectures.layers.latent_space) LatentEncoder3D (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.cait_3d.CaiTAttentionWithMLPConfig 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.Swin3DEncoderWithPatchEmbeddingsConfig attribute) length (vision_architectures.layers.embeddings.AbsolutePositionEmbeddings1DConfig attribute) LinearNoiseScheduler (class in vision_architectures.schedulers.noise) logit_scale_learnable (vision_architectures.layers.attention.Attention1DConfig attribute) loss_fn() (vision_architectures.nets.unetr_3d_decoder.UNetR3DDecoder static method)

M

make_channels_first() (in module vision_architectures.utils.rearrange) make_channels_last() (in module vision_architectures.utils.rearrange) map_mar() (in module vision_architectures.metrics.detection) 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) mean_average_precision_mean_average_recall() (in module vision_architectures.metrics.detection) mean_average_precision_recall() (in module vision_architectures.metrics.detection) MeanAveragePrecision (class in vision_architectures.metrics.detection) MeanAverageRecall (class in vision_architectures.metrics.detection) 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) Merger (class in vision_architectures.utils.splitter_merger) mlp_drop_prob (vision_architectures.blocks.transformer.Attention1DMLPConfig attribute) mlp_ratio (vision_architectures.blocks.transformer.Attention1DMLPConfig attribute) model_validate() (vision_architectures.utils.custom_base_model.CustomBaseModel class method) 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.losses.class_balanced_cross_entropy_loss vision_architectures.metrics.detection vision_architectures.nets.cait_3d vision_architectures.nets.detr_3d 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.bounding_boxes vision_architectures.utils.clamping vision_architectures.utils.custom_base_model vision_architectures.utils.ema_network 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) 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.cait_3d.CaiTStage2Config attribute) (vision_architectures.nets.vit_3d.ViT3DDecoderWithPatchEmbeddingsConfig attribute) (vision_architectures.nets.vit_3d.ViT3DEncoderWithPatchEmbeddingsConfig attribute) num_classes (vision_architectures.losses.class_balanced_cross_entropy_loss.ClassBalancedCrossEntropyLossConfig attribute) (vision_architectures.nets.detr_3d.DETRBBoxMLPConfig attribute)

num_decoder_layers (vision_architectures.nets.detr_3d.DETR3DDecoderConfig attribute) num_encoder_layers (vision_architectures.nets.detr_3d.DETR3DEncoderConfig 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) 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.detr_3d.DETR3DConfig attribute) (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.Swin3DBlockConfig attribute) (vision_architectures.nets.swin_3d.Swin3DPatchMergingConfig attribute) (vision_architectures.nets.swin_3d.Swin3DPatchSplittingConfig attribute) (vision_architectures.nets.unetr_3d_decoder.UNetR3DStageConfig attribute) out_dim_ratio (vision_architectures.nets.maxvit_3d.MaxViT3DBlockConfig attribute) (vision_architectures.nets.swin_3d.Swin3DBlockConfig property) (vision_architectures.nets.swin_3d.Swin3DPatchMergingConfig property) 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) pairwise_bbox_giou() (vision_architectures.nets.detr_3d.DETR3D static method) pairwise_bbox_iou() (vision_architectures.nets.detr_3d.DETR3D static method) parallelize_module() (in module vision_architectures.utils.pipeline_parallelism) parallelize_pipeline() (in module vision_architectures.utils.pipeline_parallelism) patch_merging (vision_architectures.nets.swin_3d.Swin3DBlockConfig attribute) (vision_architectures.nets.swinv2_3d.SwinV23DBlockConfig attribute) patch_size (vision_architectures.layers.embeddings.PatchEmbeddings3DConfig attribute) (vision_architectures.nets.swin_3d.Swin3DEncoderWithPatchEmbeddingsConfig attribute) patch_splitting (vision_architectures.nets.swin_3d.Swin3DBlockConfig attribute) (vision_architectures.nets.swinv2_3d.SwinV23DBlockConfig 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) PixelShuffleDownsample3D (class in vision_architectures.layers.scale) PixelShuffleUpsample3D (class in vision_architectures.layers.scale) populate() (vision_architectures.nets.swin_3d.Swin3DBlockConfig method) (vision_architectures.nets.swin_3d.Swin3DEncoderDecoderConfig method) populate_docstring() (in module vision_architectures.docstrings) process (vision_architectures.nets.perceiver_3d.Perceiver3DEncoderConfig attribute) proj_drop_prob (vision_architectures.layers.attention.Attention1DConfig 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) rearrange_for_sin_coefficients() (vision_architectures.layers.embeddings.RotaryPositionEmbeddings1D 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) 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) rotary_position_embeddings_config (vision_architectures.layers.attention.Attention1DConfig attribute) (vision_architectures.layers.attention.Attention3DConfig attribute) RotaryPositionEmbeddings1D (class in vision_architectures.layers.embeddings) RotaryPositionEmbeddings3D (class in vision_architectures.layers.embeddings)

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.swin_3d.Swin3DBlockConfig property) split (vision_architectures.layers.embeddings.RotaryPositionEmbeddings3DConfig attribute) 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) stage1_depth (vision_architectures.nets.cait_3d.CaiTStage1Config attribute) stage2_depth (vision_architectures.nets.cait_3d.CaiTStage2Config attribute) stages (vision_architectures.nets.swin_3d.Swin3DEncoderDecoderConfig attribute) (vision_architectures.nets.swinv2_3d.SwinV23DEncoderDecoderConfig 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) Swin3DDecoder (class in vision_architectures.nets.swin_3d) Swin3DEncoder (class in vision_architectures.nets.swin_3d) Swin3DEncoderDecoderBase (class in vision_architectures.nets.swin_3d) Swin3DEncoderWithPatchEmbeddings (class in vision_architectures.nets.swin_3d) Swin3DLayer (class in vision_architectures.nets.swin_3d) Swin3DPatchMerging (class in vision_architectures.nets.swin_3d) Swin3DPatchSplitting (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) SwinV23DEncoderDecoderBase (class in vision_architectures.nets.swinv2_3d) SwinV23DEncoderWithPatchEmbeddings (class in vision_architectures.nets.swinv2_3d) SwinV23DLayer (class in vision_architectures.nets.swinv2_3d) SwinV23DLayerLogitScale (class in vision_architectures.nets.swinv2_3d) SwinV23DPatchMerging (class in vision_architectures.nets.swinv2_3d) SwinV23DPatchSplitting (class in vision_architectures.nets.swinv2_3d) SwinV23DStage (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)

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) TransformerDecoderBlock1DConfig (in module vision_architectures.blocks.transformer) TransformerDecoderBlock3D (class in vision_architectures.blocks.transformer)

TransformerDecoderBlock3DConfig (in module vision_architectures.blocks.transformer) TransformerEncoderBlock1D (class in vision_architectures.blocks.transformer) TransformerEncoderBlock1DConfig (in module vision_architectures.blocks.transformer) TransformerEncoderBlock3D (class in vision_architectures.blocks.transformer) TransformerEncoderBlock3DConfig (in module vision_architectures.blocks.transformer) transposed (vision_architectures.blocks.cnn.CNNBlockConfig attribute)

U

UNetR3DDecoder (class in vision_architectures.nets.unetr_3d_decoder) unfold_with_roll_3d() (in module vision_architectures.nets.perceiver_3d) unparallelize_module() (in module vision_architectures.utils.pipeline_parallelism) unparallelize_pipeline() (in module vision_architectures.utils.pipeline_parallelism) UP (vision_architectures.schedulers.cyclic.Phase attribute) update_class_prevalences() (vision_architectures.losses.class_balanced_cross_entropy_loss.ClassBalancedCrossEntropyLoss method) update_decay() (vision_architectures.utils.ema_network.EMANetwork method) 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.Swin3DEncoderWithPatchEmbeddingsConfig attribute) use_ema (vision_architectures.layers.codebook.CodebookConfig property) use_relative_position_bias (vision_architectures.nets.swin_3d.Swin3DBlockConfig 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.layers.embeddings.RotaryPositionEmbeddings3DConfig method) (vision_architectures.nets.cait_3d.CaiTConfig method) (vision_architectures.nets.detr_3d.DETR3DConfig method) (vision_architectures.nets.detr_3d.DETRBBoxMLPConfig 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.Swin3DBlockConfig method) (vision_architectures.nets.swin_3d.Swin3DEncoderDecoderConfig method) (vision_architectures.nets.swin_3d.Swin3DEncoderWithPatchEmbeddingsConfig 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.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.losses.class_balanced_cross_entropy_loss module vision_architectures.metrics.detection module

vision_architectures.nets.cait_3d module vision_architectures.nets.detr_3d 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.bounding_boxes module vision_architectures.utils.clamping module vision_architectures.utils.custom_base_model module vision_architectures.utils.ema_network 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) ViT3DDecoderWithPatchEmbeddings (class in vision_architectures.nets.vit_3d) ViT3DEncoder (class in vision_architectures.nets.vit_3d) ViT3DEncoderWithPatchEmbeddings (class in vision_architectures.nets.vit_3d)

W

window_size (vision_architectures.nets.maxvit_3d.MaxViT3DBlockConfig attribute) (vision_architectures.nets.swin_3d.Swin3DBlockConfig attribute)