Index

_

__init__() (fovi.arch.alexnet.ConvBlock method) (fovi.arch.architectures.ArchitectureRegistry method) (fovi.arch.convnextv2.Block method) (fovi.arch.convnextv2.ConvNeXtV2 method) (fovi.arch.convnextv2.GRN method) (fovi.arch.convnextv2.LayerNorm method) (fovi.arch.knn.KNNConvLayer method) (fovi.arch.knn.KNNDepthwiseConvLayer method) (fovi.arch.knn.KNNDepthwiseSeparableConvLayer method) (fovi.arch.knn.KNNGetterLayer method) (fovi.arch.knn.KNNPoolingLayer method) (fovi.arch.knn_autograd.TrainingMeta method) (fovi.arch.knn_pool_cuda.PoolMeta method) (fovi.arch.knnalexnet.KNNAlexNet method) (fovi.arch.knnalexnet.KNNAlexNetBlock method) (fovi.arch.knnconvnextv2.Block method) (fovi.arch.knnconvnextv2.ConvNeXtV2 method) (fovi.arch.knnconvnextv2.GRN method) (fovi.arch.knnconvnextv2.LayerNorm method) (fovi.arch.knnresnet.KNNResNet method) (fovi.arch.knnresnet.KNNResNetBasicBlock method) (fovi.arch.knnresnet.KNNResNetBottleneck method) (fovi.arch.knnvit.FoviDinoV3RoPE method) (fovi.arch.knnvit.KNNPartitioningPatchEmbedding method) (fovi.arch.knnvit.KNNPatchEmbedding method) (fovi.arch.knnvit.KNNViT method) (fovi.arch.knnvit.PartitioningPatchEmbedding method) (fovi.arch.mlp.LayerBlock method) (fovi.arch.mlp.MLPWrapper method) (fovi.arch.norm.KNNBatchNorm method) (fovi.arch.norm.KNNChannelNorm method) (fovi.arch.norm.KNNCoordNorm method) (fovi.arch.norm.KNNGroupNorm method) (fovi.arch.norm.KNNLayerNorm method) (fovi.arch.polar.PolarPadder method) (fovi.arch.pretrained_resnet.KNNConvLayer method) (fovi.arch.pretrained_resnet.KNNResNet method) (fovi.arch.resnet.BasicBlockPolar method) (fovi.arch.resnet.BottleneckPolar method) (fovi.arch.resnet.ResNet method) (fovi.arch.vit.MultiHeadSelfAttention method) (fovi.arch.vit.PatchEmbedding method) (fovi.arch.vit.PositionalEncoding method) (fovi.arch.vit.RoPEMultiHeadSelfAttention method) (fovi.arch.vit.RoPEPositionalEncoding method) (fovi.arch.vit.RoPETransformerBlock method) (fovi.arch.vit.TransformerBlock method) (fovi.arch.vit.VisionTransformer method) (fovi.arch.wrapper.BackboneProjectorWrapper method) (fovi.fovinet.FoviNet method) (fovi.probes.FoviNetProbe method) (fovi.probes.FoviNetProbes method) (fovi.probes.LinearProbe method) (fovi.sensing.coords.SamplingCoords method) (fovi.sensing.manifold.CorticalSensorManifold method) (fovi.sensing.policies.BaseSaccadePolicy method) (fovi.sensing.policies.MultiRandomSaccadePolicy method) (fovi.sensing.policies.NoSaccadePolicy method) (fovi.sensing.policies.PolicyRegistry method) (fovi.sensing.retina.GaussianColorDecay method) (fovi.sensing.retina.RetinalTransform method) (fovi.sensing.samplers.GaussianKNNGridSampler method) (fovi.sensing.samplers.GridSampler method) (fovi.sensing.samplers.KNNGridSampler method) (fovi.trainer.Trainer method) (fovi.utils.fastaugs.transforms.CircularMask method) (fovi.utils.fastaugs.transforms.ColorJitter method) (fovi.utils.fastaugs.transforms.Compose method) (fovi.utils.fastaugs.transforms.MultiSample method) (fovi.utils.fastaugs.transforms.NormalizeGPU method) (fovi.utils.fastaugs.transforms.RandomApply method) (fovi.utils.fastaugs.transforms.RandomBrightness method) (fovi.utils.fastaugs.transforms.RandomColorJitter method) (fovi.utils.fastaugs.transforms.RandomColorJitterYIQ method) (fovi.utils.fastaugs.transforms.RandomContrast method) (fovi.utils.fastaugs.transforms.RandomGaussianBlur method) (fovi.utils.fastaugs.transforms.RandomGrayscale method) (fovi.utils.fastaugs.transforms.RandomHorizontalFlip method) (fovi.utils.fastaugs.transforms.RandomPatchShuffle method) (fovi.utils.fastaugs.transforms.RandomRotate method) (fovi.utils.fastaugs.transforms.RandomRotateObject method) (fovi.utils.fastaugs.transforms.RandomSolarization method) (fovi.utils.fastaugs.transforms.RandomZoom method) (fovi.utils.fastaugs.transforms.ToDevice method) (fovi.utils.fastaugs.transforms.ToFloat method) (fovi.utils.fastaugs.transforms.ToFloatDiv method) (fovi.utils.fastaugs.transforms.ToGrayscaleGPU method) (fovi.utils.fastaugs.transforms.ToGrayscaleTorchGPU method) (fovi.utils.fastaugs.transforms.ToTorchImage method) (fovi.utils.flops.FlopWrapper method) (fovi.utils.HiddenPrints method) (fovi.utils.lora.LoRAParam method) (fovi.utils.losses.BarlowTwinsLoss method) (fovi.utils.losses.SimCLRLoss method) (fovi.utils.losses.VicRegLoss method) (fovi.utils.lr_scheduling.CosineDecayWithWarmup method) (fovi.utils.lr_scheduling.LARS method) (fovi.visualizer.Visualizer method)

_announce_pre_transform_execution() (fovi.sensing.retina.RetinalTransform method) _apply_aspect_ratio() (fovi.sensing.retina.RetinalTransform method) _apply_local_rf() (fovi.arch.knn.KNNConvLayer method) (fovi.arch.pretrained_resnet.KNNConvLayer method) _apply_local_rf_to_weights() (fovi.arch.knn.KNNConvLayer method) (fovi.arch.pretrained_resnet.KNNConvLayer method) _as_geodesic_vertices() (fovi.arch.knn.KNNBaseLayer static method) _check_aspect_ratio() (fovi.sensing.retina.RetinalTransform method) _check_fix_loc() (fovi.sensing.retina.RetinalTransform method) _check_fixation_size() (fovi.sensing.retina.RetinalTransform method) _clear_non_serializable_cached_data() (fovi.utils.fastaugs.functional_tensor.Tensor method) _compute_all_distances() (fovi.arch.knn.KNNBaseLayer method) _compute_euclidean_distances_batched() (fovi.arch.knn.KNNBaseLayer method) _compute_gaussian_weights() (fovi.arch.knn.KNNPoolingLayer method) _compute_geodesic_distances() (fovi.arch.knn.KNNBaseLayer method) _compute_knns() (fovi.arch.knn.KNNBaseLayer method) _direct_grid() (fovi.sensing.samplers.BaseGridSampler method) _direct_pixel_coords() (fovi.sensing.samplers.BaseGridSampler method) _effective_weight_t() (in module fovi.arch.knn_autograd) _fast_pre_transforms_supported() (fovi.sensing.retina.RetinalTransform method) _get_sbc_mat() (in module fovi.utils.fastaugs.functional_tensor) _get_scale_factor() (in module fovi.utils.fastaugs.functional), [1] _grid_sample() (in module fovi.utils.fastaugs.functional) _init_conv_like() (fovi.arch.knn.KNNConvLayer method) (fovi.arch.pretrained_resnet.KNNConvLayer method) _init_depthwise_conv_like() (fovi.arch.knn.KNNDepthwiseConvLayer method) _init_depthwise_separable_conv_like() (fovi.arch.knn.KNNDepthwiseSeparableConvLayer method) _init_fix_agg() (fovi.probes.FoviNetProbe method) _init_probes() (fovi.probes.FoviNetProbes method) _init_weights() (fovi.arch.vit.VisionTransformer method) _is_gather_gemm_layer() (in module fovi.arch.knn_optimization) _make_layer() (fovi.arch.knnresnet.KNNResNet method) (fovi.arch.pretrained_resnet.KNNResNet method) _mask_invalid_outputs() (fovi.arch.knn.KNNBaseLayer method) _mask_invalid_samples() (fovi.sensing.samplers.BaseGridSampler static method) _native_cuda_supported() (in module fovi.arch.knn_optimization) _node_index() (in module fovi.arch.knn_gather_gemm) _nout_chunk() (in module fovi.arch.knn_autograd) _pad_and_gather_knns() (fovi.arch.knn.KNNConvLayer method) (fovi.arch.pretrained_resnet.KNNConvLayer method) _pad_flat_input() (in module fovi.arch.knn_autograd) _padding_mask_src() (fovi.sensing.retina.RetinalTransform method) _prep_grid_for_grid_sample() (fovi.sensing.samplers.BaseGridSampler method) _prepare_coords() (in module fovi.utils.fastaugs.functional) _prepare_mat() (in module fovi.utils.fastaugs.functional) _prepare_param() (in module fovi.utils.fastaugs.functional) _prepare_rot_pt() (in module fovi.utils.fastaugs.functional) _prepare_sampling_args() (fovi.sensing.samplers.BaseGridSampler method) _prepare_thetas() (in module fovi.utils.fastaugs.functional) _random_grayscale_points() (fovi.sensing.retina.RetinalTransform method) _register_ops() (in module fovi.arch.knn_gather_gemm) _sample_then_pre_transform() (fovi.sensing.retina.RetinalTransform method) _scatter_matrix() (in module fovi.arch.knn_gather_gemm) _set_knn_padding_metadata() (fovi.arch.knn.KNNBaseLayer method) _stable_neighbor_order() (fovi.arch.knnvit.KNNPartitioningPatchEmbedding static method) _torch_direct_sample() (fovi.sensing.samplers.GridSampler method) _torch_grid_sample() (fovi.sensing.samplers.GridSampler method) _transform_fix_grid() (fovi.sensing.samplers.BaseGridSampler method) _uint8_to_unit() (fovi.sensing.retina.RetinalTransform method)

A

A (fovi.utils.lora.LoRAParam attribute) accuracy() (fovi.utils.losses.SimCLRLoss method) add_supervised_meters() (fovi.trainer.Trainer method) add_to_all() (in module fovi.utils) adjust_brightness() (in module fovi.utils.fastaugs.functional) (in module fovi.utils.fastaugs.functional_tensor) adjust_brightness_pil() (in module fovi.utils.fastaugs.functional) adjust_contrast() (in module fovi.utils.fastaugs.functional) (in module fovi.utils.fastaugs.functional_tensor) adjust_contrast_logit() (in module fovi.utils.fastaugs.functional) adjust_contrast_pil() (in module fovi.utils.fastaugs.functional) adjust_gamma() (in module fovi.utils.fastaugs.functional_tensor) adjust_hue() (in module fovi.utils.fastaugs.functional_tensor) adjust_hue_fast() (in module fovi.utils.fastaugs.functional_tensor) adjust_rms_contrast() (in module fovi.utils.fastaugs.functional) adjust_rms_contrast_logit() (in module fovi.utils.fastaugs.functional) adjust_saturation() (in module fovi.utils.fastaugs.functional_tensor) adjust_sharpness() (in module fovi.utils.fastaugs.functional_tensor) affine() (in module fovi.utils.fastaugs.functional_tensor) affine_transform() (in module fovi.utils.fastaugs.functional) alexnet2023() (in module fovi.arch.architectures) alexnet2023_baseline() (in module fovi.arch.alexnet) all_coords() (fovi.sensing.samplers.GridSampler method) (fovi.sensing.samplers.KNNGridSampler method) alpha (fovi.utils.lora.LoRAParam attribute) analyze_rf() (in module fovi.utils) apply_2d_rotary_pos_emb() (in module fovi.arch.vit)

apply_last() (fovi.utils.fastaugs.transforms.ColorJitter method) (fovi.utils.fastaugs.transforms.Compose method) (fovi.utils.fastaugs.transforms.NormalizeGPU method) (fovi.utils.fastaugs.transforms.RandomApply method) (fovi.utils.fastaugs.transforms.RandomBrightness method) (fovi.utils.fastaugs.transforms.RandomColorJitter method) (fovi.utils.fastaugs.transforms.RandomColorJitterYIQ method) (fovi.utils.fastaugs.transforms.RandomContrast method) (fovi.utils.fastaugs.transforms.RandomGaussianBlur method) (fovi.utils.fastaugs.transforms.RandomGrayscale method) (fovi.utils.fastaugs.transforms.RandomHorizontalFlip method) (fovi.utils.fastaugs.transforms.RandomPatchShuffle method) (fovi.utils.fastaugs.transforms.RandomSolarization method) (fovi.utils.fastaugs.transforms.ToChannelsFirst method) (fovi.utils.fastaugs.transforms.ToChannelsLast method) (fovi.utils.fastaugs.transforms.ToDevice method) (fovi.utils.fastaugs.transforms.ToFloat method) (fovi.utils.fastaugs.transforms.ToFloatDiv method) (fovi.utils.fastaugs.transforms.ToNumpy method) (fovi.utils.fastaugs.transforms.ToTorchImage method) apply_lora() (in module fovi.arch.pretrained_resnet) (in module fovi.utils.lora) apply_mask() (in module fovi.utils.fastaugs.functional) arch_wrapper() (in module fovi.arch.architectures) ArchitectureRegistry (class in fovi.arch.architectures) async_copy (fovi.arch.knn_cuda.KernelConfig attribute) auto_match_num_coords() (in module fovi.sensing.coords) autocontrast() (in module fovi.utils.fastaugs.functional_tensor)

B

B (fovi.utils.lora.LoRAParam attribute) BackboneProjectorWrapper (class in fovi.arch.wrapper) backup_model() (in module fovi.utils.backup) backup_models() (in module fovi.utils.backup) backward() (fovi.arch.knn_autograd.KNNConvFunction static method) (fovi.arch.knn_pool_cuda.KNNPoolFunction static method) (fovi.utils.fastaugs.functional_tensor.Tensor method) backward_combined() (fovi.arch.knn_cuda.CudaOps static method) (in module fovi.arch.knn_cuda) BarlowTwinsLoss (class in fovi.utils.losses) BaseGridSampler (class in fovi.sensing.samplers) BaseSaccadePolicy (class in fovi.sensing.policies) BasicBlockPolar (class in fovi.arch.resnet) batch_size (fovi.utils.losses.BarlowTwinsLoss attribute) (fovi.utils.losses.SimCLRLoss attribute) before_call() (fovi.utils.fastaugs.transforms.CircularMask method) (fovi.utils.fastaugs.transforms.ColorJitter method) (fovi.utils.fastaugs.transforms.NormalizeGPU method) (fovi.utils.fastaugs.transforms.RandomApply method) (fovi.utils.fastaugs.transforms.RandomBrightness method) (fovi.utils.fastaugs.transforms.RandomColorJitter method) (fovi.utils.fastaugs.transforms.RandomColorJitterYIQ method) (fovi.utils.fastaugs.transforms.RandomContrast method) (fovi.utils.fastaugs.transforms.RandomGaussianBlur method) (fovi.utils.fastaugs.transforms.RandomGrayscale method) (fovi.utils.fastaugs.transforms.RandomHorizontalFlip method) (fovi.utils.fastaugs.transforms.RandomPatchShuffle method) (fovi.utils.fastaugs.transforms.RandomRotate method) (fovi.utils.fastaugs.transforms.RandomRotateObject method) (fovi.utils.fastaugs.transforms.RandomSolarization method) (fovi.utils.fastaugs.transforms.RandomZoom method)

bk (fovi.arch.knn_cuda.KernelConfig attribute) Block (class in fovi.arch.convnextv2) (class in fovi.arch.knnconvnextv2) blur_radius (fovi.utils.fastaugs.transforms.CircularMask property) bm (fovi.arch.knn_cuda.KernelConfig attribute) bn (fovi.arch.knn_cuda.KernelConfig attribute) (fovi.utils.losses.BarlowTwinsLoss attribute) BottleneckPolar (class in fovi.arch.resnet) build_fovi_dinov3() (in module fovi.arch.dinov3) build_model() (in module fovi.arch.alexnet)

C

cart_to_complex_log() (in module fovi.sensing.coords) cart_to_polar() (in module fovi.sensing.coords) center_crop() (in module fovi.utils.fastaugs.functional_tensor) change_sigma() (fovi.sensing.retina.RetinalTransform method) checkpoint() (fovi.trainer.Trainer method) cin (fovi.arch.knn_autograd.TrainingMeta attribute) CircularMask (class in fovi.utils.fastaugs.transforms) cleanup_activation_hooks() (fovi.fovinet.FoviNet method) cleanup_distributed() (fovi.trainer.Trainer method) clear_cache() (in module fovi.arch.knn_optimization) clear_kernel_cache() (in module fovi.arch.knn_cuda) (in module fovi.sensing.grid_sample_cuda) clear_optimized_cache() (fovi.arch.knn.KNNConvLayer method) (fovi.arch.pretrained_resnet.KNNConvLayer method) clear_pool_cache() (in module fovi.arch.knn_pool_cuda) clone() (fovi.sensing.coords.SamplingCoords method) color_jitter() (in module fovi.utils.fastaugs.functional_tensor) ColorJitter (class in fovi.utils.fastaugs.transforms) compact_forward() (in module fovi.arch.knn_autograd) CompactTorchOps (class in fovi.arch.knn_autograd) Compose (class in fovi.utils.fastaugs.transforms) compute_activations() (fovi.trainer.Trainer method) compute_binary_receptive_field() (in module fovi.arch.knn) compute_local_rf() (fovi.arch.knn.KNNConvLayer method) (fovi.arch.pretrained_resnet.KNNConvLayer method) compute_logits() (fovi.utils.losses.SimCLRLoss method) compute_receptive_field() (in module fovi.arch.knn) compute_reference_coords() (fovi.arch.knn.KNNConvLayer method) (fovi.arch.pretrained_resnet.KNNConvLayer method) conv2d() (in module fovi.utils.fastaugs.functional_tensor) conv3x3_polar() (in module fovi.arch.resnet) conv_polar() (in module fovi.arch.resnet)

ConvBlock (class in fovi.arch.alexnet) convert_image_dtype() (in module fovi.utils.fastaugs.functional_tensor) convert_model_to_polar() (in module fovi.arch.polar) convnext_pico() (in module fovi.arch.convnextv2) ConvNeXtV2 (class in fovi.arch.convnextv2) (class in fovi.arch.knnconvnextv2) convnextv2_atto() (in module fovi.arch.convnextv2) convnextv2_base() (in module fovi.arch.convnextv2) convnextv2_femto() (in module fovi.arch.convnextv2) convnextv2_huge() (in module fovi.arch.convnextv2) convnextv2_large() (in module fovi.arch.convnextv2) convnextv2_nano() (in module fovi.arch.convnextv2) convnextv2_tiny() (in module fovi.arch.convnextv2) copy_hydra_outputs() (fovi.trainer.Trainer method) cort_cartesian_to_cort_cylindrical() (fovi.sensing.manifold.CorticalSensorManifold method) cortical_cylindrical_to_cortical_cartesian() (in module fovi.sensing.manifold) CorticalSensorManifold (class in fovi.sensing.manifold) cosine_decay() (in module fovi.utils.lr_scheduling) cosine_scheduler() (in module fovi.utils.lr_scheduling) CosineDecayWithWarmup (class in fovi.utils.lr_scheduling) count (fovi.arch.knn_pool_cuda.PoolMeta attribute) cout (fovi.arch.knn_autograd.TrainingMeta attribute) cov_coeff (fovi.utils.losses.VicRegLoss attribute) create_model_and_scaler() (fovi.trainer.Trainer method) create_optimizer() (fovi.trainer.Trainer method) create_standard_loader() (fovi.trainer.Trainer method) create_train_loader() (fovi.trainer.Trainer method) create_val_loader() (fovi.trainer.Trainer method) criterion (fovi.utils.losses.SimCLRLoss attribute) crop() (in module fovi.utils.fastaugs.functional) (in module fovi.utils.fastaugs.functional_tensor) CudaOps (class in fovi.arch.knn_cuda) cylindrical_to_visual_polar() (fovi.sensing.manifold.CorticalSensorManifold method)

D

dataclass() (in module fovi.arch.knn_autograd) DEFAULT (fovi.arch.pretrained_resnet.ResNet18_Weights attribute) (torchvision.models.ResNet18_Weights attribute) default_config() (in module fovi.arch.knn_cuda) default_grad_input_config() (in module fovi.arch.knn_cuda) default_grad_weight_config() (in module fovi.arch.knn_cuda) default_grad_weight_ksplit() (in module fovi.arch.knn_cuda) detach() (fovi.utils.fastaugs.functional_tensor.Tensor method) detach_() (fovi.utils.fastaugs.functional_tensor.Tensor method) device (fovi.arch.knn_autograd.TrainingMeta attribute) (fovi.arch.knn_pool_cuda.PoolMeta attribute) dim_order() (fovi.utils.fastaugs.functional_tensor.Tensor method)

div_() (in module fovi.utils.fastaugs.functional) dm_dr() (fovi.sensing.manifold.CorticalSensorManifold method) download_model() (in module fovi.hub) draw_params() (fovi.utils.fastaugs.transforms.CircularMask method) (fovi.utils.fastaugs.transforms.RandomRotate method) (fovi.utils.fastaugs.transforms.RandomRotateObject method) (fovi.utils.fastaugs.transforms.RandomZoom method) drho_dr() (fovi.sensing.manifold.CorticalSensorManifold method) dropout (fovi.probes.FoviNetProbe attribute) (fovi.probes.FoviNetProbes attribute) (fovi.probes.LinearProbe attribute) dummy_red() (in module fovi.utils.fastaugs.functional)

E

eig() (fovi.utils.fastaugs.functional_tensor.Tensor method) end_lr (fovi.utils.lr_scheduling.CosineDecayWithWarmup attribute) ensure_pool_metadata() (in module fovi.arch.knn_pool_cuda) ensure_training_metadata() (in module fovi.arch.knn_autograd) equalize() (in module fovi.utils.fastaugs.functional_tensor)

eval_and_log() (fovi.trainer.Trainer method) exec() (fovi.trainer.Trainer class method) expansion (fovi.arch.knnresnet.KNNResNetBasicBlock attribute) (fovi.arch.knnresnet.KNNResNetBottleneck attribute) (fovi.arch.resnet.BasicBlockPolar attribute), [1] (fovi.arch.resnet.BottleneckPolar attribute), [1]

F

final_accuracy() (fovi.trainer.Trainer method) find_config() (in module fovi.trainer) find_desired_res() (in module fovi.sensing.coords) find_valid_num_groups() (in module fovi.arch.norm) five_crop() (in module fovi.utils.fastaugs.functional_tensor) fix_agg (fovi.probes.FoviNetProbe attribute) (fovi.probes.FoviNetProbes attribute) fixation_size (fovi.sensing.policies.MultiRandomSaccadePolicy attribute) flatten_dict() (in module fovi.utils) flattened_basic_blocks() (in module fovi.arch.pretrained_resnet) flip_mat() (in module fovi.utils.fastaugs.functional) FlopWrapper (class in fovi.utils.flops) forward() (fovi.arch.convnextv2.Block method) (fovi.arch.convnextv2.ConvNeXtV2 method) (fovi.arch.convnextv2.GRN method) (fovi.arch.convnextv2.LayerNorm method) (fovi.arch.knn.KNNConvLayer method) (fovi.arch.knn.KNNDepthwiseConvLayer method) (fovi.arch.knn.KNNDepthwiseSeparableConvLayer method) (fovi.arch.knn.KNNGetterLayer method) (fovi.arch.knn.KNNPoolingLayer method) (fovi.arch.knn_autograd.CompactTorchOps static method) (fovi.arch.knn_autograd.KNNConvFunction static method) (fovi.arch.knn_cuda.CudaOps static method) (fovi.arch.knn_gather_gemm.GatherGemmOps static method) (fovi.arch.knn_pool_cuda.KNNPoolFunction static method) (fovi.arch.knnalexnet.KNNAlexNet method) (fovi.arch.knnalexnet.KNNAlexNetBlock method) (fovi.arch.knnconvnextv2.Block method) (fovi.arch.knnconvnextv2.ConvNeXtV2 method) (fovi.arch.knnconvnextv2.GRN method) (fovi.arch.knnconvnextv2.LayerNorm method) (fovi.arch.knnresnet.KNNResNet method) (fovi.arch.knnresnet.KNNResNetBasicBlock method) (fovi.arch.knnresnet.KNNResNetBottleneck method) (fovi.arch.knnvit.FoviDinoV3RoPE method) (fovi.arch.knnvit.KNNPatchEmbedding method) (fovi.arch.knnvit.KNNViT method) (fovi.arch.mlp.MLPWrapper method) (fovi.arch.norm.KNNChannelNorm method) (fovi.arch.polar.PolarPadder method) (fovi.arch.pretrained_resnet.KNNConvLayer method) (fovi.arch.pretrained_resnet.KNNResNet method) (fovi.arch.resnet.BasicBlockPolar method) (fovi.arch.resnet.BottleneckPolar method) (fovi.arch.vit.MultiHeadSelfAttention method) (fovi.arch.vit.PatchEmbedding method) (fovi.arch.vit.PositionalEncoding method) (fovi.arch.vit.RoPEMultiHeadSelfAttention method) (fovi.arch.vit.RoPEPositionalEncoding method) (fovi.arch.vit.TransformerBlock method) (fovi.arch.vit.VisionTransformer method) (fovi.arch.wrapper.BackboneProjectorWrapper method) (fovi.fovinet.FoviNet method) (fovi.probes.FoviNetProbe method) (fovi.probes.FoviNetProbes method) (fovi.probes.LinearProbe method) (fovi.sensing.policies.MultiRandomSaccadePolicy method) (fovi.sensing.policies.NoSaccadePolicy method) (fovi.sensing.retina.GaussianColorDecay method) (fovi.sensing.retina.RetinalTransform method) (fovi.sensing.samplers.GridSampler method) (fovi.sensing.samplers.KNNGridSampler method) (fovi.utils.flops.FlopWrapper method) (fovi.utils.lora.LoRAParam method) (fovi.utils.losses.BarlowTwinsLoss method) (fovi.utils.losses.SimCLRLoss method) (fovi.utils.losses.VicRegLoss method) (in module fovi.arch.knn_cuda) forward_features() (fovi.arch.convnextv2.ConvNeXtV2 method) (fovi.arch.knnconvnextv2.ConvNeXtV2 method) forward_ssl() (fovi.fovinet.FoviNet method) forward_supervised() (fovi.fovinet.FoviNet method) fovi module fovi.arch module fovi.arch.alexnet module fovi.arch.architectures module fovi.arch.convnextv2 module fovi.arch.dinov3 module fovi.arch.knn module fovi.arch.knn_autograd module fovi.arch.knn_cuda module fovi.arch.knn_gather_gemm module fovi.arch.knn_optimization module fovi.arch.knn_pool_cuda module

fovi.arch.knnalexnet module fovi.arch.knnconvnextv2 module fovi.arch.knnresnet module fovi.arch.knnvit module fovi.arch.mlp module fovi.arch.norm module fovi.arch.polar module fovi.arch.pretrained_resnet module fovi.arch.resnet module fovi.arch.vit module fovi.arch.wrapper module fovi.demo module fovi.fovinet module fovi.hub module fovi.paths module fovi.probes module fovi.sensing module fovi.sensing.coords module fovi.sensing.grid_sample_cuda module fovi.sensing.manifold module fovi.sensing.policies module fovi.sensing.retina module fovi.sensing.samplers module fovi.trainer module fovi.utils module fovi.utils.backup module fovi.utils.fastaugs module fovi.utils.fastaugs.functional module fovi.utils.fastaugs.functional_tensor module fovi.utils.fastaugs.transforms module fovi.utils.flops module fovi.utils.knnprobe module fovi.utils.lora module fovi.utils.losses module fovi.utils.lr_scheduling module fovi.utils.std_transforms module fovi.visualizer module fovi_alexnet2023() (in module fovi.arch.architectures) fovi_dinov3() (in module fovi.arch.architectures) fovi_resnet() (in module fovi.arch.architectures) fovi_resnet18() (in module fovi.arch.architectures) fovi_resnet18_dwsep() (in module fovi.arch.architectures) fovi_resnet18_dwsep_lowres() (in module fovi.arch.architectures) fovi_resnet18_lora() (in module fovi.arch.architectures) fovi_resnet18_lowres() (in module fovi.arch.architectures) fovi_resnet50() (in module fovi.arch.architectures) fovi_resnet9() (in module fovi.arch.architectures) fovi_resnet9_dwsep() (in module fovi.arch.architectures) fovi_resnet9_dwsep_lowres() (in module fovi.arch.architectures) fovi_resnet9_lowres() (in module fovi.arch.architectures) fovi_vit() (in module fovi.arch.architectures) fovi_vit_base() (in module fovi.arch.architectures) fovi_vit_custom() (in module fovi.arch.architectures) fovi_vit_small() (in module fovi.arch.architectures) fovi_vit_tiny() (in module fovi.arch.architectures) FoviDinoV3RoPE (class in fovi.arch.knnvit) FoviNet (class in fovi.fovinet) FoviNetProbe (class in fovi.probes) FoviNetProbes (class in fovi.probes)

G

GatherGemmOps (class in fovi.arch.knn_gather_gemm) gaussian_blur() (in module fovi.utils.fastaugs.functional_tensor) GaussianColorDecay (class in fovi.sensing.retina) GaussianKNNGridSampler (class in fovi.sensing.samplers) generate_batch_permutations() (in module fovi.utils.fastaugs.functional) get() (fovi.arch.architectures.ArchitectureRegistry method) (fovi.sensing.policies.PolicyRegistry method) get_activations() (fovi.fovinet.FoviNet method) get_backbone() (in module fovi.arch.alexnet) get_captured_activations() (fovi.fovinet.FoviNet method) get_first_conv_weights() (in module fovi.visualizer) get_flops_df() (in module fovi.utils.flops) get_image_as_batch() (in module fovi.demo) get_image_num_channels() (in module fovi.utils.fastaugs.functional_tensor) get_image_size() (in module fovi.utils.fastaugs.functional_tensor) get_in_channels() (fovi.fovinet.FoviNet method) get_in_out_coords() (in module fovi.arch.knn) get_inputs() (fovi.utils.flops.FlopWrapper method) get_isotropic_sampling_coords() (in module fovi.sensing.coords) get_knn_conv_layer() (in module fovi.arch.knn) get_logpolar_image_sampling_coords() (in module fovi.sensing.coords) get_min_cmf_a() (in module fovi.sensing.retina) get_mlp() (in module fovi.arch.mlp) get_model() (in module fovi.utils) get_model_from_base_fn() (in module fovi) get_norm() (in module fovi.arch.norm) get_num_coords() (fovi.fovinet.FoviNet method)

get_random_crop() (fovi.sensing.policies.BaseSaccadePolicy method) get_random_name() (in module fovi.utils) get_random_nearcenter_fixation() (fovi.sensing.policies.BaseSaccadePolicy method) get_random_nearcenter_fixations_batch() (fovi.sensing.policies.BaseSaccadePolicy method) get_repr_size() (in module fovi.arch.alexnet) (in module fovi.arch.resnet) get_repr_sizes() (fovi.fovinet.FoviNet method) get_sampling_coords() (in module fovi.sensing.coords) get_scatter_sizes() (fovi.sensing.coords.SamplingCoords method) get_std_transforms() (in module fovi.utils.std_transforms) get_strided_coords() (fovi.sensing.coords.SamplingCoords method) get_trainer_from_base_fn() (in module fovi) get_transforms() (fovi.fovinet.FoviNet method) get_warp_params() (fovi.sensing.retina.RetinalTransform method) get_warped_cartesian_sampling_coords() (in module fovi.sensing.coords) grad_input() (fovi.arch.knn_autograd.CompactTorchOps static method) (fovi.arch.knn_cuda.CudaOps static method) (fovi.arch.knn_gather_gemm.GatherGemmOps static method) (in module fovi.arch.knn_cuda) grad_weight() (fovi.arch.knn_autograd.CompactTorchOps static method) (fovi.arch.knn_cuda.CudaOps static method) (fovi.arch.knn_gather_gemm.GatherGemmOps static method) (in module fovi.arch.knn_cuda) grid_sample() (in module fovi.utils.fastaugs.functional) (in module fovi.utils.fastaugs.functional_tensor) GridSampler (class in fovi.sensing.samplers) GRN (class in fovi.arch.convnextv2) (class in fovi.arch.knnconvnextv2)

H

has() (fovi.arch.architectures.ArchitectureRegistry method) (fovi.sensing.policies.PolicyRegistry method) hflip() (in module fovi.utils.fastaugs.functional_tensor) HiddenPrints (class in fovi.utils) hsv_jitter() (in module fovi.utils.fastaugs.functional) hsv_jitter2() (in module fovi.utils.fastaugs.functional)

hsv_jitter_array() (in module fovi.utils.fastaugs.functional) hsv_jitter_array2() (in module fovi.utils.fastaugs.functional) hsv_jitter_tensor() (in module fovi.utils.fastaugs.functional) hsv_mat() (in module fovi.utils.fastaugs.functional) hsv_mat2() (in module fovi.utils.fastaugs.functional) hue_mat() (in module fovi.utils.fastaugs.functional)

I

IMAGENET1K_V1 (fovi.arch.pretrained_resnet.ResNet18_Weights attribute) (torchvision.models.ResNet18_Weights attribute) index() (fovi.utils.fastaugs.functional_tensor.Tensor method) indices (fovi.arch.knn_pool_cuda.PoolMeta attribute) indices_i64 (fovi.arch.knn_pool_cuda.PoolMeta attribute) init_cortical_mesh() (fovi.sensing.manifold.CorticalSensorManifold method) init_fixation_system() (fovi.fovinet.FoviNet method) init_visual_mesh() (fovi.sensing.manifold.CorticalSensorManifold method) initialize_logger() (fovi.trainer.Trainer method) initialize_remote_logger() (fovi.trainer.Trainer method)

input_linear (fovi.arch.knn_autograd.TrainingMeta attribute) input_linear_flat (fovi.arch.knn_autograd.TrainingMeta attribute) interpolate() (in module fovi.utils.fastaugs.functional_tensor) inv_count (fovi.arch.knn_pool_cuda.PoolMeta attribute) inv_freq (fovi.arch.knnvit.FoviDinoV3RoPE attribute) invert() (in module fovi.utils.fastaugs.functional_tensor) is_conv (fovi.utils.lora.LoRAParam attribute) is_shared() (fovi.utils.fastaugs.functional_tensor.Tensor method) istft() (fovi.utils.fastaugs.functional_tensor.Tensor method) Iterable (class in fovi.arch.pretrained_resnet)

K

k (fovi.arch.knn_autograd.TrainingMeta attribute) (fovi.arch.knn_pool_cuda.PoolMeta attribute) KernelConfig (class in fovi.arch.knn_cuda) knn_classifier() (in module fovi.utils.knnprobe) knn_probe() (in module fovi.utils.knnprobe) KNNAlexNet (class in fovi.arch.knnalexnet) KNNAlexNetBlock (class in fovi.arch.knnalexnet) KNNBaseLayer (class in fovi.arch.knn) KNNBatchNorm (class in fovi.arch.norm) KNNChannelNorm (class in fovi.arch.norm) KNNConvFunction (class in fovi.arch.knn_autograd) KNNConvLayer (class in fovi.arch.knn) (class in fovi.arch.pretrained_resnet) knnconvnextv2_atto() (in module fovi.arch.knnconvnextv2) knnconvnextv2_base() (in module fovi.arch.knnconvnextv2) knnconvnextv2_femto() (in module fovi.arch.knnconvnextv2) knnconvnextv2_huge() (in module fovi.arch.knnconvnextv2) knnconvnextv2_large() (in module fovi.arch.knnconvnextv2) knnconvnextv2_nano() (in module fovi.arch.knnconvnextv2)

knnconvnextv2_pico() (in module fovi.arch.knnconvnextv2) knnconvnextv2_tiny() (in module fovi.arch.knnconvnextv2) KNNCoordNorm (class in fovi.arch.norm) KNNDepthwiseConvLayer (class in fovi.arch.knn) KNNDepthwiseSeparableConvLayer (class in fovi.arch.knn) KNNGetterLayer (class in fovi.arch.knn) KNNGridSampler (class in fovi.sensing.samplers) KNNGroupNorm (class in fovi.arch.norm) KNNLayerNorm (class in fovi.arch.norm) KNNPartitioningPatchEmbedding (class in fovi.arch.knnvit) KNNPatchEmbedding (class in fovi.arch.knnvit) KNNPoolFunction (class in fovi.arch.knn_pool_cuda) KNNPoolingLayer (class in fovi.arch.knn) KNNResNet (class in fovi.arch.knnresnet) (class in fovi.arch.pretrained_resnet) KNNResNetBasicBlock (class in fovi.arch.knnresnet) KNNResNetBottleneck (class in fovi.arch.knnresnet) KNNViT (class in fovi.arch.knnvit) ksteps (fovi.arch.knn_cuda.KernelConfig attribute) kwargs (fovi.utils.flops.FlopWrapper attribute)

L

lambd (fovi.utils.losses.BarlowTwinsLoss attribute) LARS (class in fovi.utils.lr_scheduling) last_params() (fovi.utils.fastaugs.transforms.CircularMask method) (fovi.utils.fastaugs.transforms.ColorJitter method) (fovi.utils.fastaugs.transforms.NormalizeGPU method) (fovi.utils.fastaugs.transforms.RandomApply method) (fovi.utils.fastaugs.transforms.RandomBrightness method) (fovi.utils.fastaugs.transforms.RandomColorJitter method) (fovi.utils.fastaugs.transforms.RandomColorJitterYIQ method) (fovi.utils.fastaugs.transforms.RandomContrast method) (fovi.utils.fastaugs.transforms.RandomGaussianBlur method) (fovi.utils.fastaugs.transforms.RandomGrayscale method) (fovi.utils.fastaugs.transforms.RandomHorizontalFlip method) (fovi.utils.fastaugs.transforms.RandomPatchShuffle method) (fovi.utils.fastaugs.transforms.RandomRotate method) (fovi.utils.fastaugs.transforms.RandomRotateObject method) (fovi.utils.fastaugs.transforms.RandomSolarization method) (fovi.utils.fastaugs.transforms.RandomZoom method) (fovi.utils.fastaugs.transforms.ToChannelsFirst method) (fovi.utils.fastaugs.transforms.ToChannelsLast method) (fovi.utils.fastaugs.transforms.ToDevice method) (fovi.utils.fastaugs.transforms.ToFloat method) (fovi.utils.fastaugs.transforms.ToFloatDiv method) (fovi.utils.fastaugs.transforms.ToNumpy method) (fovi.utils.fastaugs.transforms.ToTorchImage method) launch_from_args() (fovi.trainer.Trainer class method)

LayerBlock (class in fovi.arch.mlp) LayerNorm (class in fovi.arch.convnextv2) (class in fovi.arch.knnconvnextv2) learning_schedule() (in module fovi.utils.lr_scheduling) LinearProbe (class in fovi.probes) list_architectures() (fovi.arch.architectures.ArchitectureRegistry method) list_available_layers() (fovi.fovinet.FoviNet method) list_knn_layers() (fovi.fovinet.FoviNet method) lms_to_lgrby() (in module fovi.utils.fastaugs.functional) load_checkpoint() (fovi.trainer.Trainer method) load_conv2d_weights() (fovi.arch.knn.KNNConvLayer method) (fovi.arch.pretrained_resnet.KNNConvLayer method) load_conv3d_weights() (fovi.arch.knn.KNNConvLayer method) (fovi.arch.pretrained_resnet.KNNConvLayer method) load_dinov3() (in module fovi.arch.dinov3) load_image_for_sampling() (in module fovi.demo) load_pretrained() (in module fovi.utils) load_state_dict() (fovi.utils.lr_scheduling.CosineDecayWithWarmup method) load_torchvision_resnet18() (in module fovi.arch.pretrained_resnet) load_torchvision_resnet_backbone() (in module fovi.arch.pretrained_resnet) log() (fovi.trainer.Trainer method) logit() (in module fovi.utils.fastaugs.functional) logpolar_radius() (in module fovi.sensing.coords) LoRAParam (class in fovi.utils.lora) lstsq() (fovi.utils.fastaugs.functional_tensor.Tensor method) lu() (fovi.utils.fastaugs.functional_tensor.Tensor method)

M

m() (fovi.sensing.manifold.CorticalSensorManifold method) make_flop_counter() (in module fovi.utils.flops) map_3d() (fovi.sensing.manifold.CorticalSensorManifold method) map_to_xyz() (fovi.sensing.manifold.CorticalSensorManifold method) mask (fovi.utils.losses.SimCLRLoss attribute) mask_batch() (in module fovi.utils.fastaugs.functional), [1] mask_correlated_samples() (fovi.utils.losses.SimCLRLoss method) mask_tensor() (in module fovi.utils.fastaugs.functional) mat3() (in module fovi.utils.fastaugs.functional) (in module fovi.utils.fastaugs.functional_tensor) matched_resnet18() (in module fovi.arch.architectures) matched_resnet50() (in module fovi.arch.architectures) measure_latency() (in module fovi.utils.flops) min_diff_for_cmf_a() (in module fovi.sensing.retina) mlp_spec (fovi.probes.FoviNetProbes attribute) MLPWrapper (class in fovi.arch.mlp) model (fovi.visualizer.Visualizer attribute) module fovi fovi.arch fovi.arch.alexnet fovi.arch.architectures fovi.arch.convnextv2 fovi.arch.dinov3 fovi.arch.knn fovi.arch.knn_autograd fovi.arch.knn_cuda fovi.arch.knn_gather_gemm fovi.arch.knn_optimization fovi.arch.knn_pool_cuda fovi.arch.knnalexnet fovi.arch.knnconvnextv2 fovi.arch.knnresnet fovi.arch.knnvit fovi.arch.mlp fovi.arch.norm fovi.arch.polar fovi.arch.pretrained_resnet fovi.arch.resnet fovi.arch.vit fovi.arch.wrapper fovi.demo fovi.fovinet fovi.hub fovi.paths fovi.probes fovi.sensing fovi.sensing.coords fovi.sensing.grid_sample_cuda fovi.sensing.manifold fovi.sensing.policies fovi.sensing.retina fovi.sensing.samplers fovi.trainer fovi.utils fovi.utils.backup fovi.utils.fastaugs fovi.utils.fastaugs.functional fovi.utils.fastaugs.functional_tensor fovi.utils.fastaugs.transforms fovi.utils.flops fovi.utils.knnprobe fovi.utils.lora fovi.utils.losses fovi.utils.lr_scheduling fovi.utils.std_transforms fovi.visualizer

module_load() (fovi.utils.fastaugs.functional_tensor.Tensor method) multi_fixation_accuracy() (fovi.visualizer.Visualizer method) multi_policy (fovi.sensing.policies.MultiRandomSaccadePolicy attribute) MultiHeadSelfAttention (class in fovi.arch.vit) MultiRandomSaccadePolicy (class in fovi.sensing.policies) MultiSample (class in fovi.utils.fastaugs.transforms)

N

n_fixations (fovi.sensing.policies.MultiRandomSaccadePolicy attribute) name (fovi.arch.knn_autograd.CompactTorchOps attribute) (fovi.arch.knn_cuda.CudaOps attribute) (fovi.arch.knn_gather_gemm.GatherGemmOps attribute) nin (fovi.arch.knn_autograd.TrainingMeta attribute) (fovi.arch.knn_pool_cuda.PoolMeta attribute) nonrandom_val (fovi.sensing.policies.MultiRandomSaccadePolicy attribute) norm() (fovi.utils.fastaugs.functional_tensor.Tensor method) norm_dist_from_center (fovi.sensing.policies.MultiRandomSaccadePolicy attribute) normalize() (in module fovi.utils) (in module fovi.utils.fastaugs.functional), [1]

normalize_coords() (fovi.sensing.manifold.CorticalSensorManifold method) normalize_imagenet() (in module fovi.utils) NormalizeGPU (class in fovi.utils.fastaugs.transforms) NoSaccadePolicy (class in fovi.sensing.policies) nout (fovi.arch.knn_autograd.TrainingMeta attribute) (fovi.arch.knn_pool_cuda.PoolMeta attribute) num_classes (fovi.probes.FoviNetProbe attribute) (fovi.probes.FoviNetProbes attribute) num_features (fovi.probes.FoviNetProbe attribute) num_sampling_coords() (in module fovi.sensing.coords) num_sampling_coords_isotropic() (in module fovi.sensing.coords)

O

op() (fovi.utils.fastaugs.transforms.NormalizeGPU method) optimized_forward() (in module fovi.arch.knn_optimization)

optimized_pool_forward() (in module fovi.arch.knn_pool_cuda) out_base_dir (fovi.visualizer.Visualizer attribute) out_dir (fovi.visualizer.Visualizer attribute)

P

p (fovi.arch.knn_autograd.TrainingMeta attribute) p64 (fovi.arch.knn_autograd.TrainingMeta attribute) pad() (in module fovi.utils.fastaugs.functional_tensor) pad_cartesian() (fovi.sensing.coords.SamplingCoords method) pairs_per_sample (fovi.utils.losses.SimCLRLoss attribute) PartitioningPatchEmbedding (class in fovi.arch.knnvit) PatchEmbedding (class in fovi.arch.vit) perspective() (in module fovi.utils.fastaugs.functional_tensor) phi_3d() (fovi.sensing.manifold.CorticalSensorManifold method) plot_coordinates() (fovi.visualizer.Visualizer method) plot_rf_diameters() (fovi.visualizer.Visualizer method) plot_sampling_grids() (fovi.visualizer.Visualizer method) plot_spatial_rf_from_multi_layers() (fovi.visualizer.Visualizer method) polar_to_cart() (in module fovi.sensing.coords)

PolarPadder (class in fovi.arch.polar) PolicyRegistry (class in fovi.sensing.policies) pool_backward() (in module fovi.arch.knn_pool_cuda) pool_forward() (in module fovi.arch.knn_pool_cuda) pool_function() (in module fovi.arch.knn_pool_cuda) pool_meta_from_indices() (in module fovi.arch.knn_pool_cuda) PoolMeta (class in fovi.arch.knn_pool_cuda) PositionalEncoding (class in fovi.arch.vit) posterize() (in module fovi.utils.fastaugs.functional_tensor) prep_fovi_dinov3_finetuning() (in module fovi.arch.dinov3) prep_fovi_resnet_finetuning() (in module fovi.arch.pretrained_resnet) print_all_functions() (fovi.visualizer.Visualizer method) probe (fovi.probes.LinearProbe attribute) probes (fovi.probes.FoviNetProbes attribute)

Q

q (fovi.arch.knn_autograd.TrainingMeta attribute)

R

r (fovi.utils.lora.LoRAParam attribute) r_from_z() (fovi.sensing.manifold.CorticalSensorManifold method) radians() (in module fovi.utils.fastaugs.functional) random_adjust_brightness() (in module fovi.utils.fastaugs.functional) random_adjust_contrast() (in module fovi.utils.fastaugs.functional) random_choice() (fovi.utils.fastaugs.transforms.RandomPatchShuffle method) random_color_jitter() (in module fovi.utils.fastaugs.functional_tensor) random_gaussian_blur2d() (in module fovi.utils.fastaugs.functional) random_grayscale() (in module fovi.utils.fastaugs.functional) random_grid_sample() (in module fovi.utils.fastaugs.functional), [1] random_hsv_jitter2() (in module fovi.utils.fastaugs.functional) random_solarization() (in module fovi.utils.fastaugs.functional) RandomApply (class in fovi.utils.fastaugs.transforms) RandomBrightness (class in fovi.utils.fastaugs.transforms) RandomColorJitter (class in fovi.utils.fastaugs.transforms) RandomColorJitterYIQ (class in fovi.utils.fastaugs.transforms) RandomContrast (class in fovi.utils.fastaugs.transforms) RandomGaussianBlur (class in fovi.utils.fastaugs.transforms) RandomGrayscale (class in fovi.utils.fastaugs.transforms) RandomHorizontalFlip (class in fovi.utils.fastaugs.transforms) RandomPatchShuffle (class in fovi.utils.fastaugs.transforms) RandomRotate (class in fovi.utils.fastaugs.transforms) RandomRotateObject (class in fovi.utils.fastaugs.transforms) RandomSolarization (class in fovi.utils.fastaugs.transforms) RandomZoom (class in fovi.utils.fastaugs.transforms) rb (fovi.arch.knn_cuda.KernelConfig attribute) register() (fovi.arch.architectures.ArchitectureRegistry method) (fovi.sensing.policies.PolicyRegistry method) register_hook() (fovi.utils.fastaugs.functional_tensor.Tensor method) register_ops() (in module fovi.arch.knn_autograd) register_post_accumulate_grad_hook() (fovi.utils.fastaugs.functional_tensor.Tensor method) reinforce() (fovi.utils.fastaugs.functional_tensor.Tensor method) remove_lora() (in module fovi.utils.lora) replay() (fovi.utils.fastaugs.transforms.CircularMask method) (fovi.utils.fastaugs.transforms.RandomRotate method) (fovi.utils.fastaugs.transforms.RandomRotateObject method) (fovi.utils.fastaugs.transforms.RandomZoom method)

reproducible_results() (in module fovi.utils) resample_patch_embed_conv() (in module fovi.arch.knnvit) reset_model() (fovi.trainer.Trainer method) resize() (fovi.utils.fastaugs.functional_tensor.Tensor method) (in module fovi.utils.fastaugs.functional_tensor) resize_as() (fovi.utils.fastaugs.functional_tensor.Tensor method) ResNet (class in fovi.arch.resnet) resnet101() (in module fovi.arch.resnet) resnet18() (in module fovi.arch.architectures) (in module fovi.arch.pretrained_resnet) (in module fovi.arch.resnet) ResNet18_Weights (class in fovi.arch.pretrained_resnet) (class in torchvision.models) resnet34() (in module fovi.arch.architectures) (in module fovi.arch.resnet) resnet50() (in module fovi.arch.architectures) (in module fovi.arch.resnet) resnet_ssl() (in module fovi.arch.resnet) retinal_transform (fovi.sensing.policies.MultiRandomSaccadePolicy attribute) (fovi.sensing.policies.NoSaccadePolicy attribute) RetinalTransform (class in fovi.sensing.retina) rev_col (fovi.arch.knn_autograd.TrainingMeta attribute) rev_kn (fovi.arch.knn_pool_cuda.PoolMeta attribute) rev_rowptr (fovi.arch.knn_autograd.TrainingMeta attribute) (fovi.arch.knn_pool_cuda.PoolMeta attribute) reverse_map() (fovi.sensing.manifold.CorticalSensorManifold method) rgb_to_grayscale() (in module fovi.utils.fastaugs.functional_tensor) rgb_to_xyz() (in module fovi.utils.fastaugs.functional) rho_3d() (fovi.sensing.manifold.CorticalSensorManifold method) RoPEMultiHeadSelfAttention (class in fovi.arch.vit) RoPEPositionalEncoding (class in fovi.arch.vit) RoPETransformerBlock (class in fovi.arch.vit) rotate() (in module fovi.utils.fastaugs.functional) (in module fovi.utils.fastaugs.functional_tensor) rotate_mat() (in module fovi.utils.fastaugs.functional) rotate_object() (in module fovi.utils.fastaugs.functional) rotate_object_mat() (in module fovi.utils.fastaugs.functional) rowcol_to_xy() (in module fovi.sensing.coords)

S

sample_fixations() (fovi.sensing.policies.BaseSaccadePolicy method) sample_float() (in module fovi.sensing.grid_sample_cuda) sample_params() (fovi.utils.fastaugs.transforms.ColorJitter static method) (fovi.utils.fastaugs.transforms.RandomColorJitter static method) (fovi.utils.fastaugs.transforms.RandomColorJitterYIQ method) sample_uint8() (in module fovi.sensing.grid_sample_cuda) SamplingCoords (class in fovi.sensing.coords) sat_mat() (in module fovi.utils.fastaugs.functional) save_checkpoint() (fovi.trainer.Trainer method) scaled_base_lr (fovi.utils.lr_scheduling.CosineDecayWithWarmup attribute) scaling (fovi.utils.lora.LoRAParam attribute) scatter_forward() (in module fovi.arch.knn_autograd) select_backend() (in module fovi.arch.knn_optimization) set_trace() (in module fovi.utils.fastaugs.functional) setup_activation_hooks() (fovi.fovinet.FoviNet method) setup_distributed() (fovi.trainer.Trainer method) share_memory_() (fovi.utils.fastaugs.functional_tensor.Tensor method)

sigmoid() (in module fovi.utils.fastaugs.functional) sim_coeff (fovi.utils.losses.VicRegLoss attribute) SimCLRLoss (class in fovi.utils.losses) similarity_f (fovi.utils.losses.SimCLRLoss attribute) solarization() (in module fovi.utils.fastaugs.functional) solarize() (in module fovi.utils.fastaugs.functional_tensor) solve() (fovi.utils.fastaugs.functional_tensor.Tensor method) split() (fovi.utils.fastaugs.functional_tensor.Tensor method) srgb_to_lms() (in module fovi.utils.fastaugs.functional) srgb_to_lrgb() (in module fovi.utils.fastaugs.functional) state_dict() (fovi.utils.lr_scheduling.CosineDecayWithWarmup method) std_coeff (fovi.utils.losses.VicRegLoss attribute) step() (fovi.utils.lr_scheduling.LARS method) stft() (fovi.utils.fastaugs.functional_tensor.Tensor method) storage() (fovi.utils.fastaugs.functional_tensor.Tensor method) storage_type() (fovi.utils.fastaugs.functional_tensor.Tensor method) symeig() (fovi.utils.fastaugs.functional_tensor.Tensor method)

T

temperature (fovi.utils.losses.SimCLRLoss attribute) ten_crop() (in module fovi.utils.fastaugs.functional_tensor) Tensor (class in fovi.utils.fastaugs.functional_tensor) timeit() (in module fovi.utils) to() (fovi.fovinet.FoviNet method) (fovi.sensing.coords.SamplingCoords method) to_channels_first() (in module fovi.utils.fastaugs.functional), [1] to_channels_last() (in module fovi.utils.fastaugs.functional), [1] to_device() (in module fovi.utils.fastaugs.functional) to_float() (in module fovi.utils.fastaugs.functional) to_grayscale() (in module fovi.utils.fastaugs.functional) to_numpy() (in module fovi.utils.fastaugs.functional) to_sparse_coo() (fovi.utils.fastaugs.functional_tensor.Tensor method) to_torch_image() (in module fovi.utils.fastaugs.functional) ToChannelsFirst (class in fovi.utils.fastaugs.transforms) ToChannelsLast (class in fovi.utils.fastaugs.transforms) ToDevice (class in fovi.utils.fastaugs.transforms)

ToFloat (class in fovi.utils.fastaugs.transforms) ToFloatDiv (class in fovi.utils.fastaugs.transforms) ToGrayscaleGPU (class in fovi.utils.fastaugs.transforms) ToGrayscaleTorchGPU (class in fovi.utils.fastaugs.transforms) ToNumpy (class in fovi.utils.fastaugs.transforms) torch_pad() (in module fovi.utils.fastaugs.functional_tensor) ToTorchImage (class in fovi.utils.fastaugs.transforms) train() (fovi.trainer.Trainer method) train_loop() (fovi.trainer.Trainer method) Trainer (class in fovi.trainer) trainer (fovi.utils.flops.FlopWrapper attribute) (fovi.visualizer.Visualizer attribute) TrainingMeta (class in fovi.arch.knn_autograd) transform_sampling_grid() (in module fovi.sensing.coords) TransformerBlock (class in fovi.arch.vit) translate_mat() (in module fovi.utils.fastaugs.functional) trunc_normal_() (in module fovi.utils.lr_scheduling)

U

unflatten() (fovi.utils.fastaugs.functional_tensor.Tensor method) unflatten_dict() (in module fovi.utils)

unique() (fovi.utils.fastaugs.functional_tensor.Tensor method) unique_consecutive() (fovi.utils.fastaugs.functional_tensor.Tensor method)

V

v (fovi.arch.knn_autograd.TrainingMeta attribute) val_loop() (fovi.trainer.Trainer method) val_mat() (in module fovi.utils.fastaugs.functional) vec (fovi.arch.knn_cuda.KernelConfig attribute) vflip() (in module fovi.utils.fastaugs.functional_tensor) VicRegLoss (class in fovi.utils.losses) vis_cartesian_to_cort_cartesian() (fovi.sensing.manifold.CorticalSensorManifold method) vis_cartesian_to_cortical_cartesian_coords() (in module fovi.sensing.manifold) vis_cartesian_to_cortical_cylindrical() (in module fovi.sensing.manifold) VisionTransformer (class in fovi.arch.vit)

visualize_filter() (fovi.visualizer.Visualizer method) visualize_filter_over_space() (fovi.visualizer.Visualizer method) visualize_filters() (fovi.visualizer.Visualizer method) (in module fovi.visualizer) visualize_filters_knn() (in module fovi.visualizer) Visualizer (class in fovi.visualizer) vit() (in module fovi.arch.architectures) vit_base() (in module fovi.arch.architectures) vit_custom() (in module fovi.arch.architectures) vit_small() (in module fovi.arch.architectures) vit_tiny() (in module fovi.arch.architectures)

W

weight_linear (fovi.arch.knn_autograd.TrainingMeta attribute) weight_linear_flat (fovi.arch.knn_autograd.TrainingMeta attribute) wm (fovi.arch.knn_cuda.KernelConfig attribute)

wn (fovi.arch.knn_cuda.KernelConfig attribute) world_size (fovi.utils.losses.BarlowTwinsLoss attribute) (fovi.utils.losses.SimCLRLoss attribute)

X

xy_to_colrow() (in module fovi.sensing.coords)

xy_to_rowcol() (in module fovi.sensing.coords) xyz_to_lms() (in module fovi.utils.fastaugs.functional)

Z

z_3d() (fovi.sensing.manifold.CorticalSensorManifold method) z_integrand() (fovi.sensing.manifold.CorticalSensorManifold method)

zoom() (in module fovi.utils.fastaugs.functional) zoom_mat() (in module fovi.utils.fastaugs.functional)