Index

_

__init__() (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.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.wrapper.BackboneProjectorWrapper method) (fovi.models.alexnet.ConvBlock method) (fovi.models.architectures.ArchitectureRegistry method) (fovi.models.convnextv2.Block method) (fovi.models.convnextv2.ConvNeXtV2 method) (fovi.models.convnextv2.GRN method) (fovi.models.convnextv2.LayerNorm method) (fovi.models.fovinet.FoviNet method) (fovi.models.knnalexnet.KNNAlexNet method) (fovi.models.knnalexnet.KNNAlexNetBlock method) (fovi.models.knnconvnextv2.Block method) (fovi.models.knnconvnextv2.ConvNeXtV2 method) (fovi.models.knnconvnextv2.GRN method) (fovi.models.knnconvnextv2.LayerNorm method) (fovi.models.knnresnet.KNNResNet method) (fovi.models.knnresnet.KNNResNetBasicBlock method) (fovi.models.knnresnet.KNNResNetBottleneck method) (fovi.models.knnvit.FoviDinoV3RoPE method) (fovi.models.knnvit.KNNPartitioningPatchEmbedding method) (fovi.models.knnvit.KNNPatchEmbedding method) (fovi.models.knnvit.KNNViT method) (fovi.models.knnvit.PartitioningPatchEmbedding method) (fovi.models.loading.DictConfig method) (fovi.models.loading.GlobalHydra method) (fovi.models.loading.HiddenPrints method) (fovi.models.loading.OmegaConf method) (fovi.models.pretrained_resnet.KNNConvLayer method) (fovi.models.pretrained_resnet.KNNResNet method) (fovi.models.probes.FoviNetProbe method) (fovi.models.probes.FoviNetProbes method) (fovi.models.probes.LinearProbe method) (fovi.models.resnet.BasicBlockPolar method) (fovi.models.resnet.BottleneckPolar method) (fovi.models.resnet.ResNet method) (fovi.models.vit.MultiHeadSelfAttention method) (fovi.models.vit.PatchEmbedding method) (fovi.models.vit.PositionalEncoding method) (fovi.models.vit.RoPEMultiHeadSelfAttention method) (fovi.models.vit.RoPEPositionalEncoding method) (fovi.models.vit.RoPETransformerBlock method) (fovi.models.vit.TransformerBlock method) (fovi.models.vit.VisionTransformer 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.training.loader.FlashLoader method) (fovi.training.loading.HiddenPrints method) (fovi.training.loading.OmegaConf method) (fovi.training.loading.Trainer method) (fovi.training.trainer.Trainer method) (fovi.training.utils.flops.FlopWrapper method) (fovi.training.utils.losses.BarlowTwinsLoss method) (fovi.training.utils.losses.SimCLRLoss method) (fovi.training.utils.losses.VicRegLoss method) (fovi.training.utils.lr_scheduling.CosineDecayWithWarmup method) (fovi.training.utils.lr_scheduling.LARS method) (fovi.training.visualizer.Visualizer method) (fovi.utils.fastaugs.loader.FlashLoader 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.models.pretrained_resnet.KNNConvLayer method) _apply_local_rf_to_weights() (fovi.arch.knn.KNNConvLayer method) (fovi.models.pretrained_resnet.KNNConvLayer method) _as_geodesic_vertices() (fovi.arch.knn.KNNBaseLayer static method) _canary_grad_stage_ok() (in module fovi.arch.knn_warp) _canary_launch_ok() (in module fovi.arch.knn_warp) _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_grad_kernel() (in module fovi.arch.knn_warp) _get_sbc_mat() (in module fovi.utils.fastaugs.functional_tensor) _get_scale_factor() (in module fovi.utils.fastaugs.functional), [1] _get_uncached_batch_config() (in module fovi.arch.knn_warp) _grid_sample() (in module fovi.utils.fastaugs.functional) _init_conv_like() (fovi.arch.knn.KNNConvLayer method) (fovi.models.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.models.probes.FoviNetProbe method) _init_probes() (fovi.models.probes.FoviNetProbes method) _init_weights() (fovi.models.vit.VisionTransformer method) _is_gather_gemm_layer() (in module fovi.arch.knn_optimization) _make_grad_input_stage2_kernel() (in module fovi.arch.knn_warp) _make_grad_weight_stage2_kernel() (in module fovi.arch.knn_warp) _make_grad_weight_stage3_kernel() (in module fovi.arch.knn_warp) _make_grad_weight_stage4_kernel() (in module fovi.arch.knn_warp) _make_layer() (fovi.models.knnresnet.KNNResNet method) (fovi.models.pretrained_resnet.KNNResNet method) _make_uncached_r2_kernel() (in module fovi.arch.knn_warp) _make_uncached_r4_kernel() (in module fovi.arch.knn_warp) _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.models.pretrained_resnet.KNNConvLayer method) _pad_bias_to_tiles() (in module fovi.arch.knn_warp) _pad_flat_input() (in module fovi.arch.knn_autograd) _pad_flat_input_fp16() (in module fovi.arch.knn_warp) _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) _promote() (fovi.models.loading.DictConfig method) _random_grayscale_points() (fovi.sensing.retina.RetinalTransform method) _register_ops() (in module fovi.arch.knn_gather_gemm) (in module fovi.arch.knn_warp) _sample_then_pre_transform() (fovi.sensing.retina.RetinalTransform method) _scatter_matrix() (in module fovi.arch.knn_gather_gemm) _select_uncached_batch_config() (in module fovi.arch.knn_warp) _set_knn_padding_metadata() (fovi.arch.knn.KNNBaseLayer method) _stable_neighbor_order() (fovi.models.knnvit.KNNPartitioningPatchEmbedding static method) _to_object() (fovi.models.loading.DictConfig 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) absolute() (fovi.models.loading.Path method) accuracy() (fovi.training.utils.losses.SimCLRLoss method) (fovi.utils.losses.SimCLRLoss method) add_supervised_meters() (fovi.training.loading.Trainer method) (fovi.training.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.models.architectures) alexnet2023_baseline() (in module fovi.models.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.models.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.models.pretrained_resnet) (in module fovi.utils.lora) apply_mask() (in module fovi.utils.fastaugs.functional) arch_wrapper() (in module fovi.models.architectures) ArchitectureRegistry (class in fovi.models.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.training.utils.backup) (in module fovi.utils.backup) backup_models() (in module fovi.training.utils.backup) (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.training.utils.losses) (class in fovi.utils.losses) BaseGridSampler (class in fovi.sensing.samplers) BaseSaccadePolicy (class in fovi.sensing.policies) BasicBlockPolar (class in fovi.models.resnet) batch_size (fovi.training.utils.losses.BarlowTwinsLoss attribute) (fovi.training.utils.losses.SimCLRLoss attribute) (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.models.convnextv2) (class in fovi.models.knnconvnextv2) blur_radius (fovi.utils.fastaugs.transforms.CircularMask property) bm (fovi.arch.knn_cuda.KernelConfig attribute) bn (fovi.arch.knn_cuda.KernelConfig attribute) (fovi.training.utils.losses.BarlowTwinsLoss attribute) (fovi.utils.losses.BarlowTwinsLoss attribute) BottleneckPolar (class in fovi.models.resnet) build_fovi_dinov3() (in module fovi.models.dinov3) build_model() (in module fovi.models.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.training.loading.Trainer method) (fovi.training.trainer.Trainer method) chmod() (fovi.models.loading.Path method) cholesky() (fovi.utils.fastaugs.functional_tensor.Tensor method) cin (fovi.arch.knn_autograd.TrainingMeta attribute) CircularMask (class in fovi.utils.fastaugs.transforms) cleanup_activation_hooks() (fovi.models.fovinet.FoviNet method) cleanup_distributed() (fovi.training.loading.Trainer method) (fovi.training.trainer.Trainer method) clear() (fovi.models.loading.GlobalHydra method) clear_cache() (fovi.models.loading.OmegaConf static method) (fovi.training.loading.OmegaConf static method) (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.models.pretrained_resnet.KNNConvLayer method) clear_pool_cache() (in module fovi.arch.knn_pool_cuda) clear_resolver() (fovi.models.loading.OmegaConf class method) (fovi.training.loading.OmegaConf class method) clear_resolvers() (fovi.models.loading.OmegaConf static method) (fovi.training.loading.OmegaConf static method) 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.training.loading.Trainer method) (fovi.training.trainer.Trainer method) compute_binary_receptive_field() (in module fovi.arch.knn) compute_local_rf() (fovi.arch.knn.KNNConvLayer method) (fovi.models.pretrained_resnet.KNNConvLayer method) compute_logits() (fovi.training.utils.losses.SimCLRLoss method) (fovi.utils.losses.SimCLRLoss method) compute_receptive_field() (in module fovi.arch.knn) compute_reference_coords() (fovi.arch.knn.KNNConvLayer method) (fovi.models.pretrained_resnet.KNNConvLayer method) config_loader() (fovi.models.loading.GlobalHydra method) conv2d() (in module fovi.utils.fastaugs.functional_tensor) conv3x3_polar() (in module fovi.models.resnet) conv_polar() (in module fovi.models.resnet) ConvBlock (class in fovi.models.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.models.convnextv2) ConvNeXtV2 (class in fovi.models.convnextv2) (class in fovi.models.knnconvnextv2) convnextv2_atto() (in module fovi.models.convnextv2) convnextv2_base() (in module fovi.models.convnextv2) convnextv2_femto() (in module fovi.models.convnextv2) convnextv2_huge() (in module fovi.models.convnextv2) convnextv2_large() (in module fovi.models.convnextv2) convnextv2_nano() (in module fovi.models.convnextv2) convnextv2_tiny() (in module fovi.models.convnextv2) copy() (fovi.models.loading.DictConfig method) copy_cache() (fovi.models.loading.OmegaConf static method) (fovi.training.loading.OmegaConf static method) copy_hydra_outputs() (fovi.training.loading.Trainer method) (fovi.training.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.training.utils.lr_scheduling) (in module fovi.utils.lr_scheduling) cosine_scheduler() (in module fovi.training.utils.lr_scheduling) (in module fovi.utils.lr_scheduling) CosineDecayWithWarmup (class in fovi.training.utils.lr_scheduling) (class in fovi.utils.lr_scheduling) count (fovi.arch.knn_pool_cuda.PoolMeta attribute) count() (fovi.models.loading.Sequence method) (fovi.training.loading.Sequence method) cout (fovi.arch.knn_autograd.TrainingMeta attribute) cov_coeff (fovi.training.utils.losses.VicRegLoss attribute) (fovi.utils.losses.VicRegLoss attribute) create() (fovi.models.loading.OmegaConf static method) (fovi.training.loading.OmegaConf static method) create_model_and_scaler() (fovi.training.loading.Trainer method) (fovi.training.trainer.Trainer method) create_optimizer() (fovi.training.loading.Trainer method) (fovi.training.trainer.Trainer method) create_standard_loader() (fovi.training.loading.Trainer method) (fovi.training.trainer.Trainer method) create_train_loader() (fovi.training.loading.Trainer method) (fovi.training.trainer.Trainer method) create_val_loader() (fovi.training.loading.Trainer method) (fovi.training.trainer.Trainer method) criterion (fovi.training.utils.losses.SimCLRLoss attribute) (fovi.utils.losses.SimCLRLoss attribute) crop() (in module fovi.utils.fastaugs.functional) (in module fovi.utils.fastaugs.functional_tensor) crop_frame() (in module fovi.utils.image) CudaOps (class in fovi.arch.knn_cuda) cwd() (fovi.models.loading.Path class method) cylindrical_to_visual_polar() (fovi.sensing.manifold.CorticalSensorManifold method)

D

dataclass() (in module fovi.arch.knn_autograd) DEFAULT (fovi.models.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) default_model_dirs() (in module fovi.models.loading) 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) DictConfig (class in fovi.models.loading)

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.models.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.models.probes.FoviNetProbe attribute) (fovi.models.probes.FoviNetProbes attribute) (fovi.models.probes.LinearProbe attribute) dummy_red() (in module fovi.utils.fastaugs.functional)

E

eig() (fovi.utils.fastaugs.functional_tensor.Tensor method) end_lr (fovi.training.utils.lr_scheduling.CosineDecayWithWarmup attribute) (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.training.loading.Trainer method) (fovi.training.trainer.Trainer method)

exec() (fovi.training.loading.Trainer class method) (fovi.training.trainer.Trainer class method) exists() (fovi.models.loading.Path method) expanduser() (fovi.models.loading.Path method) expansion (fovi.models.knnresnet.KNNResNetBasicBlock attribute) (fovi.models.knnresnet.KNNResNetBottleneck attribute) (fovi.models.resnet.BasicBlockPolar attribute), [1] (fovi.models.resnet.BottleneckPolar attribute), [1]

F

fig_to_frame() (in module fovi.utils.image) final_accuracy() (fovi.training.loading.Trainer method) (fovi.training.trainer.Trainer method) find_config() (in module fovi.models.loading) (in module fovi.training.loading) 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.models.probes.FoviNetProbe attribute) (fovi.models.probes.FoviNetProbes attribute) fixation_size (fovi.sensing.policies.MultiRandomSaccadePolicy attribute) FlashLoader (class in fovi.training.loader) (class in fovi.utils.fastaugs.loader) flatten_dict() (in module fovi.utils) flattened_basic_blocks() (in module fovi.models.pretrained_resnet) flip_mat() (in module fovi.utils.fastaugs.functional) FlopWrapper (class in fovi.training.utils.flops) (class in fovi.utils.flops) forward() (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.knn_warp.WarpCompactOps static method) (fovi.arch.knn_warp.WarpTrainOps static method) (fovi.arch.mlp.MLPWrapper method) (fovi.arch.norm.KNNChannelNorm method) (fovi.arch.polar.PolarPadder method) (fovi.arch.wrapper.BackboneProjectorWrapper method) (fovi.models.convnextv2.Block method) (fovi.models.convnextv2.ConvNeXtV2 method) (fovi.models.convnextv2.GRN method) (fovi.models.convnextv2.LayerNorm method) (fovi.models.fovinet.FoviNet method) (fovi.models.knnalexnet.KNNAlexNet method) (fovi.models.knnalexnet.KNNAlexNetBlock method) (fovi.models.knnconvnextv2.Block method) (fovi.models.knnconvnextv2.ConvNeXtV2 method) (fovi.models.knnconvnextv2.GRN method) (fovi.models.knnconvnextv2.LayerNorm method) (fovi.models.knnresnet.KNNResNet method) (fovi.models.knnresnet.KNNResNetBasicBlock method) (fovi.models.knnresnet.KNNResNetBottleneck method) (fovi.models.knnvit.FoviDinoV3RoPE method) (fovi.models.knnvit.KNNPatchEmbedding method) (fovi.models.knnvit.KNNViT method) (fovi.models.pretrained_resnet.KNNConvLayer method) (fovi.models.pretrained_resnet.KNNResNet method) (fovi.models.probes.FoviNetProbe method) (fovi.models.probes.FoviNetProbes method) (fovi.models.probes.LinearProbe method) (fovi.models.resnet.BasicBlockPolar method) (fovi.models.resnet.BottleneckPolar method) (fovi.models.vit.MultiHeadSelfAttention method) (fovi.models.vit.PatchEmbedding method) (fovi.models.vit.PositionalEncoding method) (fovi.models.vit.RoPEMultiHeadSelfAttention method) (fovi.models.vit.RoPEPositionalEncoding method) (fovi.models.vit.TransformerBlock method) (fovi.models.vit.VisionTransformer 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.training.utils.flops.FlopWrapper method) (fovi.training.utils.losses.BarlowTwinsLoss method) (fovi.training.utils.losses.SimCLRLoss method) (fovi.training.utils.losses.VicRegLoss 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.models.convnextv2.ConvNeXtV2 method) (fovi.models.knnconvnextv2.ConvNeXtV2 method) forward_ssl() (fovi.models.fovinet.FoviNet method) forward_supervised() (fovi.models.fovinet.FoviNet method) fovi module fovi.arch 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.knn_warp module fovi.arch.mlp module fovi.arch.norm module fovi.arch.polar module fovi.arch.wrapper module fovi.demo module fovi.models module fovi.models.alexnet module fovi.models.architectures module fovi.models.convnextv2 module

fovi.models.dinov3 module fovi.models.fovinet module fovi.models.hub module fovi.models.knnalexnet module fovi.models.knnconvnextv2 module fovi.models.knnresnet module fovi.models.knnvit module fovi.models.loading module fovi.models.pretrained_resnet module fovi.models.probes module fovi.models.resnet module fovi.models.vit module fovi.paths 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.training module fovi.training.loader module fovi.training.loading module fovi.training.trainer module fovi.training.utils module fovi.training.utils.backup module fovi.training.utils.flops module fovi.training.utils.knnprobe module fovi.training.utils.losses module fovi.training.utils.lr_scheduling module fovi.training.visualizer 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.loader module fovi.utils.fastaugs.transforms module fovi.utils.flops module fovi.utils.image 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.models.architectures) fovi_dinov3() (in module fovi.models.architectures) fovi_resnet() (in module fovi.models.architectures) fovi_resnet18() (in module fovi.models.architectures) fovi_resnet18_dwsep() (in module fovi.models.architectures) fovi_resnet18_dwsep_lowres() (in module fovi.models.architectures) fovi_resnet18_lora() (in module fovi.models.architectures) fovi_resnet18_lowres() (in module fovi.models.architectures) fovi_resnet50() (in module fovi.models.architectures) fovi_resnet9() (in module fovi.models.architectures) fovi_resnet9_dwsep() (in module fovi.models.architectures) fovi_resnet9_dwsep_lowres() (in module fovi.models.architectures) fovi_resnet9_lowres() (in module fovi.models.architectures) fovi_vit() (in module fovi.models.architectures) fovi_vit_base() (in module fovi.models.architectures) fovi_vit_custom() (in module fovi.models.architectures) fovi_vit_small() (in module fovi.models.architectures) fovi_vit_tiny() (in module fovi.models.architectures) FoviDinoV3RoPE (class in fovi.models.knnvit) FoviNet (class in fovi.models.fovinet) FoviNetProbe (class in fovi.models.probes) FoviNetProbes (class in fovi.models.probes) from_cli() (fovi.models.loading.OmegaConf static method) (fovi.training.loading.OmegaConf static method) from_dotlist() (fovi.models.loading.OmegaConf static method) (fovi.training.loading.OmegaConf static method)

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.models.architectures.ArchitectureRegistry method) (fovi.models.loading.DictConfig method) (fovi.sensing.policies.PolicyRegistry method) get_activations() (fovi.models.fovinet.FoviNet method) get_backbone() (in module fovi.models.alexnet) get_cache() (fovi.models.loading.OmegaConf static method) (fovi.training.loading.OmegaConf static method) get_captured_activations() (fovi.models.fovinet.FoviNet method) get_first_conv_weights() (in module fovi.training.visualizer) (in module fovi.visualizer) get_flops_df() (in module fovi.training.utils.flops) (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.models.fovinet.FoviNet method) get_in_out_coords() (in module fovi.arch.knn) get_inputs() (fovi.training.utils.flops.FlopWrapper method) (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.models.loading) get_norm() (in module fovi.arch.norm) get_num_coords() (fovi.models.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.models.alexnet) (in module fovi.models.resnet) get_repr_sizes() (fovi.models.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.training.loading) get_transforms() (fovi.models.fovinet.FoviNet method) get_type() (fovi.models.loading.OmegaConf static method) (fovi.training.loading.OmegaConf static method) get_warp_params() (fovi.sensing.retina.RetinalTransform method) get_warped_cartesian_sampling_coords() (in module fovi.sensing.coords) glob() (fovi.models.loading.Path method) GlobalHydra (class in fovi.models.loading) grad_input() (fovi.arch.knn_autograd.CompactTorchOps static method) (fovi.arch.knn_cuda.CudaOps static method) (fovi.arch.knn_gather_gemm.GatherGemmOps static method) (fovi.arch.knn_warp.WarpCompactOps static method) (fovi.arch.knn_warp.WarpTrainOps static method) (in module fovi.arch.knn_cuda) (in module fovi.arch.knn_warp) grad_weight() (fovi.arch.knn_autograd.CompactTorchOps static method) (fovi.arch.knn_cuda.CudaOps static method) (fovi.arch.knn_gather_gemm.GatherGemmOps static method) (fovi.arch.knn_warp.WarpCompactOps static method) (fovi.arch.knn_warp.WarpTrainOps static method) (in module fovi.arch.knn_cuda) (in module fovi.arch.knn_warp) 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.models.convnextv2) (class in fovi.models.knnconvnextv2) group() (fovi.models.loading.Path method)

H

hardlink_to() (fovi.models.loading.Path method) has() (fovi.models.architectures.ArchitectureRegistry method) (fovi.sensing.policies.PolicyRegistry method) has_resolver() (fovi.models.loading.OmegaConf class method) (fovi.training.loading.OmegaConf class method) hflip() (in module fovi.utils.fastaugs.functional_tensor) HiddenPrints (class in fovi.models.loading) (class in fovi.training.loading) (class in fovi.utils)

home() (fovi.models.loading.Path class method) 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.models.pretrained_resnet.ResNet18_Weights attribute) (torchvision.models.ResNet18_Weights attribute) index() (fovi.models.loading.Sequence method) (fovi.training.loading.Sequence method) (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.models.fovinet.FoviNet method) init_visual_mesh() (fovi.sensing.manifold.CorticalSensorManifold method) initialize() (fovi.models.loading.GlobalHydra method) initialize_logger() (fovi.training.loading.Trainer method) (fovi.training.trainer.Trainer method) initialize_remote_logger() (fovi.training.loading.Trainer method) (fovi.training.trainer.Trainer method) input_linear (fovi.arch.knn_autograd.TrainingMeta attribute) input_linear_flat (fovi.arch.knn_autograd.TrainingMeta attribute) instance() (fovi.models.loading.GlobalHydra static method) interpolate() (in module fovi.utils.fastaugs.functional_tensor) inv_count (fovi.arch.knn_pool_cuda.PoolMeta attribute) inv_freq (fovi.models.knnvit.FoviDinoV3RoPE attribute) invert() (in module fovi.utils.fastaugs.functional_tensor) is_block_device() (fovi.models.loading.Path method) is_char_device() (fovi.models.loading.Path method) is_config() (fovi.models.loading.OmegaConf static method) (fovi.training.loading.OmegaConf static method)

is_conv (fovi.utils.lora.LoRAParam attribute) is_dict() (fovi.models.loading.OmegaConf static method) (fovi.training.loading.OmegaConf static method) is_dir() (fovi.models.loading.Path method) is_fifo() (fovi.models.loading.Path method) is_file() (fovi.models.loading.Path method) is_initialized() (fovi.models.loading.GlobalHydra method) is_interpolation() (fovi.models.loading.OmegaConf static method) (fovi.training.loading.OmegaConf static method) is_list() (fovi.models.loading.OmegaConf static method) (fovi.training.loading.OmegaConf static method) is_missing() (fovi.models.loading.OmegaConf static method) (fovi.training.loading.OmegaConf static method) is_mount() (fovi.models.loading.Path method) is_readonly() (fovi.models.loading.OmegaConf static method) (fovi.training.loading.OmegaConf static method) is_shared() (fovi.utils.fastaugs.functional_tensor.Tensor method) is_socket() (fovi.models.loading.Path method) is_struct() (fovi.models.loading.OmegaConf static method) (fovi.training.loading.OmegaConf static method) is_symlink() (fovi.models.loading.Path method) istft() (fovi.utils.fastaugs.functional_tensor.Tensor method) items() (fovi.models.loading.DictConfig method) items_ex() (fovi.models.loading.DictConfig method) Iterable (class in fovi.models.pretrained_resnet) iterdir() (fovi.models.loading.Path method)

K

k (fovi.arch.knn_autograd.TrainingMeta attribute) (fovi.arch.knn_pool_cuda.PoolMeta attribute) KernelConfig (class in fovi.arch.knn_cuda) keys() (fovi.models.loading.DictConfig method) knn_classifier() (in module fovi.training.utils.knnprobe) (in module fovi.utils.knnprobe) knn_probe() (in module fovi.training.utils.knnprobe) (in module fovi.utils.knnprobe) KNNAlexNet (class in fovi.models.knnalexnet) KNNAlexNetBlock (class in fovi.models.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.models.pretrained_resnet) knnconvnextv2_atto() (in module fovi.models.knnconvnextv2) knnconvnextv2_base() (in module fovi.models.knnconvnextv2) knnconvnextv2_femto() (in module fovi.models.knnconvnextv2) knnconvnextv2_huge() (in module fovi.models.knnconvnextv2) knnconvnextv2_large() (in module fovi.models.knnconvnextv2)

knnconvnextv2_nano() (in module fovi.models.knnconvnextv2) knnconvnextv2_pico() (in module fovi.models.knnconvnextv2) knnconvnextv2_tiny() (in module fovi.models.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.models.knnvit) KNNPatchEmbedding (class in fovi.models.knnvit) KNNPoolFunction (class in fovi.arch.knn_pool_cuda) KNNPoolingLayer (class in fovi.arch.knn) KNNResNet (class in fovi.models.knnresnet) (class in fovi.models.pretrained_resnet) KNNResNetBasicBlock (class in fovi.models.knnresnet) KNNResNetBottleneck (class in fovi.models.knnresnet) KNNViT (class in fovi.models.knnvit) ksteps (fovi.arch.knn_cuda.KernelConfig attribute) kwargs (fovi.training.utils.flops.FlopWrapper attribute) (fovi.utils.flops.FlopWrapper attribute)

L

lambd (fovi.training.utils.losses.BarlowTwinsLoss attribute) (fovi.utils.losses.BarlowTwinsLoss attribute) LARS (class in fovi.training.utils.lr_scheduling) (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.training.loading.Trainer class method) (fovi.training.trainer.Trainer class method) LayerBlock (class in fovi.arch.mlp) LayerNorm (class in fovi.models.convnextv2) (class in fovi.models.knnconvnextv2) lchmod() (fovi.models.loading.Path method) learning_schedule() (in module fovi.training.utils.lr_scheduling) (in module fovi.utils.lr_scheduling)

legacy_register_resolver() (fovi.models.loading.OmegaConf static method) (fovi.training.loading.OmegaConf static method) LinearProbe (class in fovi.models.probes) link_to() (fovi.models.loading.Path method) list_architectures() (fovi.models.architectures.ArchitectureRegistry method) list_available_layers() (fovi.models.fovinet.FoviNet method) list_knn_layers() (fovi.models.fovinet.FoviNet method) lms_to_lgrby() (in module fovi.utils.fastaugs.functional) load() (fovi.models.loading.OmegaConf static method) (fovi.training.loading.OmegaConf static method) load_checkpoint() (fovi.training.loading.Trainer method) (fovi.training.trainer.Trainer method) load_config() (in module fovi.models.loading) load_conv2d_weights() (fovi.arch.knn.KNNConvLayer method) (fovi.models.pretrained_resnet.KNNConvLayer method) load_conv3d_weights() (fovi.arch.knn.KNNConvLayer method) (fovi.models.pretrained_resnet.KNNConvLayer method) load_dinov3() (in module fovi.models.dinov3) load_image_for_sampling() (in module fovi.demo) load_pretrained() (in module fovi.utils) load_sharded_state_dict() (in module fovi.models.loading) load_state_dict() (fovi.training.utils.lr_scheduling.CosineDecayWithWarmup method) (fovi.utils.lr_scheduling.CosineDecayWithWarmup method) load_torchvision_resnet18() (in module fovi.models.pretrained_resnet) load_torchvision_resnet_backbone() (in module fovi.models.pretrained_resnet) log() (fovi.training.loading.Trainer method) (fovi.training.trainer.Trainer method) logit() (in module fovi.utils.fastaugs.functional) logpolar_radius() (in module fovi.sensing.coords) LoRAParam (class in fovi.utils.lora) lstat() (fovi.models.loading.Path method) 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.training.utils.flops) (in module fovi.utils.flops) map_3d() (fovi.sensing.manifold.CorticalSensorManifold method) map_to_xyz() (fovi.sensing.manifold.CorticalSensorManifold method) mask (fovi.training.utils.losses.SimCLRLoss attribute) (fovi.utils.losses.SimCLRLoss attribute) mask_batch() (in module fovi.utils.fastaugs.functional), [1] mask_correlated_samples() (fovi.training.utils.losses.SimCLRLoss method) (fovi.utils.losses.SimCLRLoss method) mask_tensor() (in module fovi.utils.fastaugs.functional) masked_copy() (fovi.models.loading.OmegaConf static method) (fovi.training.loading.OmegaConf static method) mat3() (in module fovi.utils.fastaugs.functional) (in module fovi.utils.fastaugs.functional_tensor) matched_resnet18() (in module fovi.models.architectures) matched_resnet50() (in module fovi.models.architectures) measure_latency() (in module fovi.training.utils.flops) (in module fovi.utils.flops) merge() (fovi.models.loading.OmegaConf static method) (fovi.training.loading.OmegaConf static method) min_diff_for_cmf_a() (in module fovi.sensing.retina) missing_keys() (fovi.models.loading.OmegaConf static method) (fovi.training.loading.OmegaConf static method) mkdir() (fovi.models.loading.Path method) mlp_spec (fovi.models.probes.FoviNetProbes attribute) MLPWrapper (class in fovi.arch.mlp) model (fovi.training.visualizer.Visualizer attribute) (fovi.visualizer.Visualizer attribute) module fovi fovi.arch 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.knn_warp fovi.arch.mlp fovi.arch.norm fovi.arch.polar fovi.arch.wrapper fovi.demo fovi.models fovi.models.alexnet fovi.models.architectures fovi.models.convnextv2 fovi.models.dinov3 fovi.models.fovinet fovi.models.hub fovi.models.knnalexnet fovi.models.knnconvnextv2 fovi.models.knnresnet fovi.models.knnvit fovi.models.loading fovi.models.pretrained_resnet fovi.models.probes fovi.models.resnet fovi.models.vit fovi.paths 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.training fovi.training.loader fovi.training.loading fovi.training.trainer fovi.training.utils fovi.training.utils.backup fovi.training.utils.flops fovi.training.utils.knnprobe fovi.training.utils.losses fovi.training.utils.lr_scheduling fovi.training.visualizer fovi.utils fovi.utils.backup fovi.utils.fastaugs fovi.utils.fastaugs.functional fovi.utils.fastaugs.functional_tensor fovi.utils.fastaugs.loader fovi.utils.fastaugs.transforms fovi.utils.flops fovi.utils.image 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.training.visualizer.Visualizer method) (fovi.visualizer.Visualizer method) multi_policy (fovi.sensing.policies.MultiRandomSaccadePolicy attribute) MultiHeadSelfAttention (class in fovi.models.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) (fovi.arch.knn_warp.WarpCompactOps attribute) (fovi.arch.knn_warp.WarpTrainOps 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.models.probes.FoviNetProbe attribute) (fovi.models.probes.FoviNetProbes attribute) num_features (fovi.models.probes.FoviNetProbe attribute) num_sampling_coords() (in module fovi.sensing.coords) num_sampling_coords_isotropic() (in module fovi.sensing.coords)

O

OmegaConf (class in fovi.models.loading) (class in fovi.training.loading) op() (fovi.utils.fastaugs.transforms.NormalizeGPU method) open() (fovi.models.loading.Path method) open_dict() (in module fovi.models.loading) (in module fovi.training.loading)

optimized_forward() (in module fovi.arch.knn_optimization) optimized_pool_forward() (in module fovi.arch.knn_pool_cuda) out_base_dir (fovi.training.visualizer.Visualizer attribute) (fovi.visualizer.Visualizer attribute) out_dir (fovi.training.visualizer.Visualizer attribute) (fovi.visualizer.Visualizer attribute) owner() (fovi.models.loading.Path method)

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.training.utils.losses.SimCLRLoss attribute) (fovi.utils.losses.SimCLRLoss attribute) PartitioningPatchEmbedding (class in fovi.models.knnvit) PatchEmbedding (class in fovi.models.vit) Path (class in fovi.models.loading) perspective() (in module fovi.utils.fastaugs.functional_tensor) phi_3d() (fovi.sensing.manifold.CorticalSensorManifold method) plot_coordinates() (fovi.training.visualizer.Visualizer method) (fovi.visualizer.Visualizer method) plot_rf_diameters() (fovi.training.visualizer.Visualizer method) (fovi.visualizer.Visualizer method) plot_sampling_grids() (fovi.training.visualizer.Visualizer method) (fovi.visualizer.Visualizer method) plot_spatial_rf_from_multi_layers() (fovi.training.visualizer.Visualizer method) (fovi.visualizer.Visualizer method)

plotly_fig_to_frame() (in module fovi.utils.image) 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) pop() (fovi.models.loading.DictConfig method) PositionalEncoding (class in fovi.models.vit) posterize() (in module fovi.utils.fastaugs.functional_tensor) prep_fovi_dinov3_finetuning() (in module fovi.models.dinov3) prep_fovi_resnet_finetuning() (in module fovi.models.pretrained_resnet) print_all_functions() (fovi.training.visualizer.Visualizer method) (fovi.visualizer.Visualizer method) probe (fovi.models.probes.LinearProbe attribute) probes (fovi.models.probes.FoviNetProbes attribute)

Q

q (fovi.arch.knn_autograd.TrainingMeta attribute)

qr() (fovi.utils.fastaugs.functional_tensor.Tensor method)

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) read_bytes() (fovi.models.loading.Path method) read_text() (fovi.models.loading.Path method) readlink() (fovi.models.loading.Path method) register() (fovi.models.architectures.ArchitectureRegistry method) (fovi.sensing.policies.PolicyRegistry method) register_hook() (fovi.utils.fastaugs.functional_tensor.Tensor method) register_new_resolver() (fovi.models.loading.OmegaConf static method) (fovi.training.loading.OmegaConf static method) register_ops() (in module fovi.arch.knn_autograd) register_post_accumulate_grad_hook() (fovi.utils.fastaugs.functional_tensor.Tensor method) register_resolver() (fovi.models.loading.OmegaConf static method) (fovi.training.loading.OmegaConf static method) reinforce() (fovi.utils.fastaugs.functional_tensor.Tensor method) remove_lora() (in module fovi.utils.lora) rename() (fovi.models.loading.Path method) replace() (fovi.models.loading.Path method) 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.models.knnvit) reset_model() (fovi.training.loading.Trainer method) (fovi.training.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) resize_image_fixed_aspect() (in module fovi.utils.image) ResNet (class in fovi.models.resnet) resnet101() (in module fovi.models.resnet) resnet18() (in module fovi.models.architectures) (in module fovi.models.pretrained_resnet) (in module fovi.models.resnet) ResNet18_Weights (class in fovi.models.pretrained_resnet) (class in torchvision.models) resnet34() (in module fovi.models.architectures) (in module fovi.models.resnet) resnet50() (in module fovi.models.architectures) (in module fovi.models.resnet) resnet_ssl() (in module fovi.models.resnet) resolve() (fovi.models.loading.OmegaConf static method) (fovi.models.loading.Path method) (fovi.training.loading.OmegaConf static method) 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) rglob() (fovi.models.loading.Path method) rho_3d() (fovi.sensing.manifold.CorticalSensorManifold method) rmdir() (fovi.models.loading.Path method) RoPEMultiHeadSelfAttention (class in fovi.models.vit) RoPEPositionalEncoding (class in fovi.models.vit) RoPETransformerBlock (class in fovi.models.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) run_cached() (in module fovi.arch.knn_warp) run_uncached() (in module fovi.arch.knn_warp)

S

samefile() (fovi.models.loading.Path method) 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() (fovi.models.loading.OmegaConf static method) (fovi.training.loading.OmegaConf static method) save_checkpoint() (fovi.training.loading.Trainer method) (fovi.training.trainer.Trainer method) save_frames_as_video() (in module fovi.utils.image) scaled_base_lr (fovi.training.utils.lr_scheduling.CosineDecayWithWarmup attribute) (fovi.utils.lr_scheduling.CosineDecayWithWarmup attribute) scaling (fovi.utils.lora.LoRAParam attribute) scatter_forward() (in module fovi.arch.knn_autograd) select() (fovi.models.loading.OmegaConf static method) (fovi.training.loading.OmegaConf static method) select_backend() (in module fovi.arch.knn_optimization) Sequence (class in fovi.models.loading) (class in fovi.training.loading) set_cache() (fovi.models.loading.OmegaConf static method) (fovi.training.loading.OmegaConf static method) set_instance() (fovi.models.loading.GlobalHydra static method) set_readonly() (fovi.models.loading.OmegaConf static method) (fovi.training.loading.OmegaConf static method) set_struct() (fovi.models.loading.OmegaConf static method) (fovi.training.loading.OmegaConf static method) set_trace() (in module fovi.utils.fastaugs.functional)

setdefault() (fovi.models.loading.DictConfig method) setup_activation_hooks() (fovi.models.fovinet.FoviNet method) setup_distributed() (fovi.training.loading.Trainer method) (fovi.training.trainer.Trainer method) share_memory_() (fovi.utils.fastaugs.functional_tensor.Tensor method) sigmoid() (in module fovi.utils.fastaugs.functional) sim_coeff (fovi.training.utils.losses.VicRegLoss attribute) (fovi.utils.losses.VicRegLoss attribute) SimCLRLoss (class in fovi.training.utils.losses) (class in fovi.utils.losses) similarity_f (fovi.training.utils.losses.SimCLRLoss attribute) (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) stat() (fovi.models.loading.Path method) state_dict() (fovi.training.utils.lr_scheduling.CosineDecayWithWarmup method) (fovi.utils.lr_scheduling.CosineDecayWithWarmup method) std_coeff (fovi.training.utils.losses.VicRegLoss attribute) (fovi.utils.losses.VicRegLoss attribute) step() (fovi.training.utils.lr_scheduling.LARS method) (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) structured() (fovi.models.loading.OmegaConf static method) (fovi.training.loading.OmegaConf static method) symeig() (fovi.utils.fastaugs.functional_tensor.Tensor method) symlink_to() (fovi.models.loading.Path method)

T

temperature (fovi.training.utils.losses.SimCLRLoss attribute) (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.models.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_container() (fovi.models.loading.OmegaConf static method) (fovi.training.loading.OmegaConf static method) 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_object() (fovi.models.loading.OmegaConf static method) (fovi.training.loading.OmegaConf static method) to_sparse_coo() (fovi.utils.fastaugs.functional_tensor.Tensor method) to_torch_image() (in module fovi.utils.fastaugs.functional) to_yaml() (fovi.models.loading.OmegaConf static method) (fovi.training.loading.OmegaConf static method) 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) touch() (fovi.models.loading.Path method) train() (fovi.training.loading.Trainer method) (fovi.training.trainer.Trainer method) train_loop() (fovi.training.loading.Trainer method) (fovi.training.trainer.Trainer method) Trainer (class in fovi.training.loading) (class in fovi.training.trainer) trainer (fovi.training.utils.flops.FlopWrapper attribute) (fovi.training.visualizer.Visualizer attribute) (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.models.vit) translate_mat() (in module fovi.utils.fastaugs.functional) trunc_normal_() (in module fovi.training.utils.lr_scheduling) (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)

unlink() (fovi.models.loading.Path method) unsafe_merge() (fovi.models.loading.OmegaConf static method) (fovi.training.loading.OmegaConf static method) update() (fovi.models.loading.OmegaConf static method) (fovi.training.loading.OmegaConf static method)

V

v (fovi.arch.knn_autograd.TrainingMeta attribute) val_loop() (fovi.training.loading.Trainer method) (fovi.training.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.training.utils.losses) (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.models.vit) visualize_filter() (fovi.training.visualizer.Visualizer method) (fovi.visualizer.Visualizer method)

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

W

WarpCompactOps (class in fovi.arch.knn_warp) WarpTrainOps (class in fovi.arch.knn_warp) 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.training.utils.losses.BarlowTwinsLoss attribute) (fovi.training.utils.losses.SimCLRLoss attribute) (fovi.utils.losses.BarlowTwinsLoss attribute) (fovi.utils.losses.SimCLRLoss attribute) write_bytes() (fovi.models.loading.Path method) write_text() (fovi.models.loading.Path method)

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)