Index

_

__call__() (vissl.data.ssl_transforms.img_patches_tensor.ImgPatchesFromTensor method) (vissl.data.ssl_transforms.shuffle_img_patches.ShuffleImgPatches method) (vissl.data.ssl_transforms.SSLTransformsWrapper method) __deepcopy__() (vissl.utils.hydra_config.AttrDict method) __delattr__() (vissl.utils.hydra_config.AttrDict method) __getattr__() (vissl.utils.hydra_config.AttrDict method) __getitem__() (vissl.data.disk_dataset.DiskImageDataset method) (vissl.data.GenericSSLDataset method) (vissl.data.ssl_dataset.GenericSSLDataset method) (vissl.data.synthetic_dataset.SyntheticImageDataset method) __getstate__() (vissl.utils.hydra_config.AttrDict method) __init__() (vissl.data.data_helper.StatefulDistributedSampler method) (vissl.data.ssl_transforms.img_pil_color_distortion.ImgPilColorDistortion method) (vissl.data.ssl_transforms.img_pil_gaussian_blur.ImgPilGaussianBlur method) (vissl.data.ssl_transforms.img_pil_multicrop_random_apply.ImgPilMultiCropRandomApply method) (vissl.data.ssl_transforms.img_pil_random_color_jitter.ImgPilRandomColorJitter method) (vissl.data.ssl_transforms.img_pil_random_photometric.ImgPilRandomPhotometric method) (vissl.data.ssl_transforms.img_pil_random_solarize.ImgPilRandomSolarize method) (vissl.data.ssl_transforms.img_pil_to_multicrop.ImgPilToMultiCrop method) (vissl.data.ssl_transforms.img_pil_to_patches_and_image.ImgPilToPatchesAndImage method) (vissl.data.ssl_transforms.img_replicate_pil.ImgReplicatePil method) (vissl.data.ssl_transforms.img_rotate_pil.ImgRotatePil method) (vissl.data.ssl_transforms.pil_photometric_transforms_lib.RandomPosterizeTransform method) (vissl.data.ssl_transforms.pil_photometric_transforms_lib.RandomSharpnessTransform method) (vissl.data.ssl_transforms.pil_photometric_transforms_lib.RandomSolarizeTransform method) (vissl.data.ssl_transforms.pil_photometric_transforms_lib.RandomValueApplier method) (vissl.data.ssl_transforms.shuffle_img_patches.ShuffleImgPatches method) (vissl.data.ssl_transforms.SSLTransformsWrapper method) (vissl.hooks.log_hooks.LogLossLrEtaHook method) (vissl.hooks.log_hooks.LogPerfTimeMetricsHook method) (vissl.hooks.swav_momentum_hooks.SwAVMomentumHook method) (vissl.hooks.tensorboard_hook.SSLTensorboardHook method) (vissl.losses.bce_logits_multiple_output_single_target.BCELogitsMultipleOutputSingleTargetLoss method) (vissl.models.heads.LinearEvalMLP method) (vissl.models.heads.MLP method) (vissl.models.heads.SiameseConcatView method) (vissl.models.heads.SwAVPrototypesHead method) (vissl.utils.hydra_config.AttrDict method)

__len__() (vissl.data.disk_dataset.DiskImageDataset method) (vissl.data.GenericSSLDataset method) (vissl.data.ssl_dataset.GenericSSLDataset method) (vissl.data.synthetic_dataset.SyntheticImageDataset method) __setattr__() (vissl.utils.hydra_config.AttrDict method) __setstate__() (vissl.utils.hydra_config.AttrDict method)

A

AccuracyListMeter (class in vissl.meters.accuracy_list_meter) add_bias_channel() (in module vissl.utils.instance_retrieval_utils.data_util) aggregate_stats() (vissl.utils.svm_utils.svm_low_shot_trainer.SVMLowShotTrainer method) AliasMethod (class in vissl.losses.nce_loss) apex (vissl.models.model_helpers.SyncBNTypes attribute) append_module_prefix() (in module vissl.utils.checkpoint) apply() (vissl.utils.instance_retrieval_utils.pca.PCA method)

apply_img_transform() (vissl.utils.instance_retrieval_utils.data_util.InstanceRetrievalImageLoader method) assert_hydra_conf() (in module vissl.utils.hydra_config) assert_learning_rate() (in module vissl.utils.hydra_config) assert_losses() (in module vissl.utils.hydra_config) AttrDict (class in vissl.utils.hydra_config) AutoContrast() (in module vissl.data.ssl_transforms.pil_photometric_transforms_lib) AutoContrastTransform (class in vissl.data.ssl_transforms.pil_photometric_transforms_lib)

B

BaseSSLMultiInputOutputModel (class in vissl.models) BatchNorm (vissl.models.model_helpers.RESNET_NORM_LAYER attribute) BCELogitsMultipleOutputSingleTargetLoss (class in vissl.losses.bce_logits_multiple_output_single_target)

build_dataloaders() (vissl.trainer.train_task.SelfSupervisionTask method) build_datasets() (vissl.trainer.train_task.SelfSupervisionTask method) build_model() (in module vissl.models) build_task() (in module vissl.trainer.trainer_main)

C

cache_url() (in module vissl.utils.io) calculate_ap() (in module vissl.utils.svm_utils.evaluate) check_data_exists() (in module vissl.data.dataset_catalog) check_model_compatibilty() (in module vissl.utils.checkpoint) CheckNanLossHook (class in vissl.hooks.state_update_hooks) checkpoint_trunk() (in module vissl.utils.activation_checkpointing) cleanup_dir() (in module vissl.utils.io) clear() (vissl.data.dataset_catalog.VisslDatasetCatalog static method) (vissl.data.VisslDatasetCatalog static method) cluster_memory() (vissl.losses.deepclusterv2_loss.DeepClusterV2Loss method) ClusterMemoryHook (class in vissl.hooks.deepclusterv2_hooks) collect_env_info() (in module vissl.utils.collect_env) compute_ap() (in module vissl.utils.instance_retrieval_utils.evaluate) compute_map() (in module vissl.utils.instance_retrieval_utils.evaluate) compute_partition_function() (vissl.losses.nce_loss.NCEAverage method)

compute_queue_scores() (vissl.losses.swav_loss.SwAVCriterion method) (vissl.losses.swav_momentum_loss.SwAVMomentumLoss method) concat_all_gather() (in module vissl.utils.misc) construct_sample_for_model() (in module vissl.trainer.train_steps.standard_train_step) convert_sync_bn() (in module vissl.models) (in module vissl.models.model_helpers) convert_to_attrdict() (in module vissl.utils.hydra_config) convert_to_one_hot() (in module vissl.data.collators.targets_one_hot_default_collator) copy_data() (in module vissl.utils.io) copy_data_to_local() (in module vissl.utils.io) copy_dir() (in module vissl.utils.io) copy_file() (in module vissl.utils.io) CosineWarmRestartScheduler (class in vissl.optimizers.param_scheduler.cosine_warm_restart_scheduler) CosineWaveTypes (class in vissl.optimizers.param_scheduler.cosine_warm_restart_scheduler) create_file_symlink() (in module vissl.utils.io) CrossEntropyMultipleOutputSingleTargetLoss (class in vissl.losses.cross_entropy_multiple_output_single_target)

D

DataloaderSyncGPUWrapper (class in vissl.data.dataloader_sync_gpu_wrapper) DeepClusterV2Loss (class in vissl.losses.deepclusterv2_loss) default_hook_generator() (in module vissl.hooks) defaults() (vissl.losses.moco_loss.MoCoLossConfig static method) DiskImageDataset (class in vissl.data.disk_dataset)

distributed_sinkhornknopp() (vissl.losses.swav_loss.SwAVCriterion method) (vissl.losses.swav_momentum_loss.SwAVMomentumLoss method) do_negative_sampling() (vissl.losses.nce_loss.NCEAverage method) draw() (vissl.losses.nce_loss.AliasMethod method) (vissl.losses.nce_loss.NumpySampler method)

E

EMA_FACTOR (vissl.utils.perf_stats.PerfMetric attribute) enable_manual_gradient_reduction() (vissl.trainer.train_task.SelfSupervisionTask property)

eval_from_ranks() (vissl.utils.instance_retrieval_utils.data_util.InstreDataset method) extract() (vissl.trainer.trainer_main.SelfSupervisionTrainer method) extract_main() (in module vissl.engines.extract_features)

F

find_free_tcp_port() (in module vissl.utils.misc) fit() (vissl.utils.instance_retrieval_utils.pca.PCA method) Flatten (class in vissl.models.model_helpers) flatten() (in module vissl.utils.instance_retrieval_utils.data_util) flops() (vissl.models.model_helpers.Flatten method) forward() (vissl.losses.bce_logits_multiple_output_single_target.BCELogitsMultipleOutputSingleTargetLoss method) (vissl.losses.cross_entropy_multiple_output_single_target.CrossEntropyMultipleOutputSingleTargetLoss method) (vissl.losses.deepclusterv2_loss.DeepClusterV2Loss method) (vissl.losses.moco_loss.MoCoLoss method) (vissl.losses.multicrop_simclr_info_nce_loss.MultiCropSimclrInfoNCECriterion method) (vissl.losses.nce_loss.NCEAverage method) (vissl.losses.nce_loss.NCECriterion method) (vissl.losses.nce_loss.NCELossWithMemory method) (vissl.losses.simclr_info_nce_loss.SimclrInfoNCECriterion method) (vissl.losses.simclr_info_nce_loss.SimclrInfoNCELoss method) (vissl.losses.swav_loss.SwAVCriterion method) (vissl.losses.swav_loss.SwAVLoss method) (vissl.losses.swav_momentum_loss.SwAVMomentumLoss method) (vissl.models.BaseSSLMultiInputOutputModel method) (vissl.models.heads.LinearEvalMLP method) (vissl.models.heads.MLP method) (vissl.models.heads.SiameseConcatView method) (vissl.models.heads.SwAVPrototypesHead method) (vissl.models.model_helpers.Flatten method) (vissl.models.model_helpers.Identity method) (vissl.models.model_helpers.Wrap method) freeze_head() (vissl.models.BaseSSLMultiInputOutputModel method) freeze_head_and_trunk() (vissl.models.BaseSSLMultiInputOutputModel method) freeze_trunk() (vissl.models.BaseSSLMultiInputOutputModel method) FreezeParametersHook (class in vissl.hooks.state_update_hooks)

from_config() (vissl.data.ssl_transforms.img_patches_tensor.ImgPatchesFromTensor class method) (vissl.data.ssl_transforms.img_pil_color_distortion.ImgPilColorDistortion class method) (vissl.data.ssl_transforms.img_pil_gaussian_blur.ImgPilGaussianBlur class method) (vissl.data.ssl_transforms.img_pil_multicrop_random_apply.ImgPilMultiCropRandomApply class method) (vissl.data.ssl_transforms.img_pil_random_color_jitter.ImgPilRandomColorJitter class method) (vissl.data.ssl_transforms.img_pil_random_photometric.ImgPilRandomPhotometric class method) (vissl.data.ssl_transforms.img_pil_random_solarize.ImgPilRandomSolarize class method) (vissl.data.ssl_transforms.img_pil_to_lab_tensor.ImgPil2LabTensor class method) (vissl.data.ssl_transforms.img_pil_to_multicrop.ImgPilToMultiCrop class method) (vissl.data.ssl_transforms.img_pil_to_patches_and_image.ImgPilToPatchesAndImage class method) (vissl.data.ssl_transforms.img_pil_to_raw_tensor.ImgPilToRawTensor class method) (vissl.data.ssl_transforms.img_replicate_pil.ImgReplicatePil class method) (vissl.data.ssl_transforms.img_rotate_pil.ImgRotatePil class method) (vissl.data.ssl_transforms.shuffle_img_patches.ShuffleImgPatches class method) (vissl.data.ssl_transforms.SSLTransformsWrapper class method) (vissl.losses.bce_logits_multiple_output_single_target.BCELogitsMultipleOutputSingleTargetLoss class method) (vissl.losses.cross_entropy_multiple_output_single_target.CrossEntropyMultipleOutputSingleTargetLoss class method) (vissl.losses.deepclusterv2_loss.DeepClusterV2Loss class method) (vissl.losses.moco_loss.MoCoLoss class method) (vissl.losses.nce_loss.NCELossWithMemory class method) (vissl.losses.simclr_info_nce_loss.SimclrInfoNCELoss class method) (vissl.losses.swav_loss.SwAVLoss class method) (vissl.losses.swav_momentum_loss.SwAVMomentumLoss class method) (vissl.meters.accuracy_list_meter.AccuracyListMeter class method) (vissl.meters.mean_ap_list_meter.MeanAPListMeter class method) (vissl.meters.mean_ap_meter.MeanAPMeter class method) (vissl.optimizers.param_scheduler.cosine_warm_restart_scheduler.CosineWarmRestartScheduler class method) (vissl.optimizers.param_scheduler.inverse_sqrt_decay.InverseSqrtScheduler class method) (vissl.trainer.train_task.SelfSupervisionTask class method) full (vissl.optimizers.param_scheduler.cosine_warm_restart_scheduler.CosineWaveTypes attribute)

G

gather_embeddings() (vissl.losses.simclr_info_nce_loss.SimclrInfoNCECriterion static method) gather_scores() (vissl.meters.mean_ap_meter.MeanAPMeter method) gather_targets() (vissl.meters.mean_ap_meter.MeanAPMeter method) gem() (in module vissl.utils.instance_retrieval_utils.data_util) GenericSSLDataset (class in vissl.data) (class in vissl.data.ssl_dataset) get() (vissl.data.dataset_catalog.VisslDatasetCatalog static method) (vissl.data.VisslDatasetCatalog static method) get_available_splits() (vissl.data.GenericSSLDataset method) (vissl.data.ssl_dataset.GenericSSLDataset method) get_avg() (vissl.utils.perf_stats.PerfMetric method) get_batchsize_per_replica() (vissl.data.GenericSSLDataset method) (vissl.data.ssl_dataset.GenericSSLDataset method) get_best_cost_value() (vissl.utils.svm_utils.svm_trainer.SVMTrainer method) get_checkpoint_folder() (in module vissl.utils.checkpoint) get_checkpoint_model_state_dict() (in module vissl.utils.checkpoint) get_checkpoint_resume_files() (in module vissl.utils.checkpoint) get_classy_state() (vissl.meters.accuracy_list_meter.AccuracyListMeter method) (vissl.meters.mean_ap_list_meter.MeanAPListMeter method) (vissl.meters.mean_ap_meter.MeanAPMeter method) (vissl.models.BaseSSLMultiInputOutputModel method) get_collator() (in module vissl.data.collators) get_config() (vissl.trainer.train_task.SelfSupervisionTask method) get_data_files() (in module vissl.data) (in module vissl.data.dataset_catalog) get_dist_run_id() (in module vissl.utils.misc) get_features() (vissl.models.BaseSSLMultiInputOutputModel method) get_file_size() (in module vissl.utils.io) get_filename() (vissl.utils.instance_retrieval_utils.data_util.InstanceRetrievalDataset method) (vissl.utils.instance_retrieval_utils.data_util.InstreDataset method) (vissl.utils.instance_retrieval_utils.data_util.RevisitedInstanceRetrievalDataset method) (vissl.utils.instance_retrieval_utils.data_util.WhiteningTrainingImageDataset method) get_global_batchsize() (vissl.data.GenericSSLDataset method) (vissl.data.ssl_dataset.GenericSSLDataset method) (vissl.trainer.train_task.SelfSupervisionTask method) get_image_paths() (vissl.data.disk_dataset.DiskImageDataset method) (vissl.data.GenericSSLDataset method) (vissl.data.ssl_dataset.GenericSSLDataset method)

get_indices_sparse() (in module vissl.utils.misc) get_json_data_catalog_file() (in module vissl.utils.misc) get_local_output_filepaths() (in module vissl.data.dataset_catalog) get_local_path() (in module vissl.data.dataset_catalog) get_machine_local_and_dist_rank() (in module vissl.utils.env) get_mean_image() (in module vissl.data.data_helper) get_model_head() (in module vissl.models.heads) get_model_trunk() (in module vissl.models.trunks) get_node_id() (in module vissl.utils.slurm) get_num_images() (vissl.utils.instance_retrieval_utils.data_util.InstanceRetrievalDataset method) (vissl.utils.instance_retrieval_utils.data_util.InstreDataset method) (vissl.utils.instance_retrieval_utils.data_util.RevisitedInstanceRetrievalDataset method) (vissl.utils.instance_retrieval_utils.data_util.WhiteningTrainingImageDataset method) get_num_query_images() (vissl.utils.instance_retrieval_utils.data_util.InstanceRetrievalDataset method) (vissl.utils.instance_retrieval_utils.data_util.InstreDataset method) (vissl.utils.instance_retrieval_utils.data_util.RevisitedInstanceRetrievalDataset method) get_optimizer_param_groups() (in module vissl.optimizers) (in module vissl.optimizers.optimizer_helper) get_precision_recall() (in module vissl.utils.svm_utils.evaluate) get_query_filename() (vissl.utils.instance_retrieval_utils.data_util.InstanceRetrievalDataset method) (vissl.utils.instance_retrieval_utils.data_util.InstreDataset method) (vissl.utils.instance_retrieval_utils.data_util.RevisitedInstanceRetrievalDataset method) get_query_roi() (vissl.utils.instance_retrieval_utils.data_util.InstanceRetrievalDataset method) (vissl.utils.instance_retrieval_utils.data_util.InstreDataset method) (vissl.utils.instance_retrieval_utils.data_util.RevisitedInstanceRetrievalDataset method) get_resume_checkpoint() (in module vissl.utils.checkpoint) get_rmac_descriptors() (in module vissl.utils.instance_retrieval_utils.rmac) get_rmac_region_coordinates() (in module vissl.utils.instance_retrieval_utils.rmac) get_scaled_lr_scheduler() (in module vissl.utils.hydra_config) get_slurm_dir() (in module vissl.utils.slurm) get_tensorboard_dir() (in module vissl.utils.tensorboard) get_tensorboard_hook() (in module vissl.utils.tensorboard) get_train_step() (in module vissl.trainer.train_steps) get_transform() (in module vissl.data.ssl_transforms) get_trunk_forward_outputs() (in module vissl.models.model_helpers) get_trunk_forward_outputs_module_list() (in module vissl.models.model_helpers) get_trunk_output_feature_names() (in module vissl.models.model_helpers)

H

half (vissl.optimizers.param_scheduler.cosine_warm_restart_scheduler.CosineWaveTypes attribute) has_checkpoint() (in module vissl.utils.checkpoint) has_data() (vissl.data.dataset_catalog.VisslDatasetCatalog static method) (vissl.data.VisslDatasetCatalog static method)

has_final_checkpoint() (in module vissl.utils.checkpoint) heads_forward() (vissl.models.BaseSSLMultiInputOutputModel method)

I

Identity (class in vissl.models.model_helpers) ImgPatchesFromTensor (class in vissl.data.ssl_transforms.img_patches_tensor) ImgPil2LabTensor (class in vissl.data.ssl_transforms.img_pil_to_lab_tensor) ImgPilColorDistortion (class in vissl.data.ssl_transforms.img_pil_color_distortion) ImgPilGaussianBlur (class in vissl.data.ssl_transforms.img_pil_gaussian_blur) ImgPilMultiCropRandomApply (class in vissl.data.ssl_transforms.img_pil_multicrop_random_apply) ImgPilRandomColorJitter (class in vissl.data.ssl_transforms.img_pil_random_color_jitter) ImgPilRandomPhotometric (class in vissl.data.ssl_transforms.img_pil_random_photometric) ImgPilRandomSolarize (class in vissl.data.ssl_transforms.img_pil_random_solarize) ImgPilToMultiCrop (class in vissl.data.ssl_transforms.img_pil_to_multicrop) ImgPilToPatchesAndImage (class in vissl.data.ssl_transforms.img_pil_to_patches_and_image) ImgPilToRawTensor (class in vissl.data.ssl_transforms.img_pil_to_raw_tensor) ImgReplicatePil (class in vissl.data.ssl_transforms.img_replicate_pil) ImgRotatePil (class in vissl.data.ssl_transforms.img_rotate_pil) ImgToTensor (class in vissl.data.ssl_transforms.img_pil_to_tensor) init_memory() (vissl.losses.deepclusterv2_loss.DeepClusterV2Loss method) (vissl.losses.nce_loss.NCEAverage method) init_model_from_weights() (in module vissl.utils.checkpoint) initialize_queue() (vissl.losses.swav_loss.SwAVCriterion method) (vissl.losses.swav_momentum_loss.SwAVMomentumLoss method)

InitMemoryHook (class in vissl.hooks.deepclusterv2_hooks) input_shape() (vissl.models.BaseSSLMultiInputOutputModel property) InstanceRetrievalDataset (class in vissl.utils.instance_retrieval_utils.data_util) InstanceRetrievalImageLoader (class in vissl.utils.instance_retrieval_utils.data_util) InstreDataset (class in vissl.utils.instance_retrieval_utils.data_util) InverseSqrtScheduler (class in vissl.optimizers.param_scheduler.inverse_sqrt_decay) is_apex_available() (in module vissl.utils.misc) is_checkpoint_phase() (in module vissl.utils.checkpoint) is_faiss_available() (in module vissl.utils.misc) is_feature_extractor_model() (in module vissl.models) (in module vissl.models.model_helpers) is_fully_frozen_model() (vissl.models.BaseSSLMultiInputOutputModel method) is_hydra_available() (in module vissl.utils.hydra_config) is_instre_dataset() (in module vissl.utils.instance_retrieval_utils.data_util) is_opencv_available() (in module vissl.utils.misc) is_revisited_dataset() (in module vissl.utils.instance_retrieval_utils.data_util) is_tensorboard_available() (in module vissl.utils.tensorboard) is_training_finished() (in module vissl.utils.checkpoint) is_url() (in module vissl.utils.io) is_whiten_dataset() (in module vissl.utils.instance_retrieval_utils.data_util)

L

l2n() (in module vissl.utils.instance_retrieval_utils.data_util) layer_splittable_before() (in module vissl.utils.activation_checkpointing) LayerNorm (vissl.models.model_helpers.RESNET_NORM_LAYER attribute) LayerNorm2d (class in vissl.models.model_helpers) LinearEvalMLP (class in vissl.models.heads) list() (vissl.data.dataset_catalog.VisslDatasetCatalog static method) (vissl.data.VisslDatasetCatalog static method) load() (vissl.utils.instance_retrieval_utils.data_util.InstanceRetrievalDataset method) load_and_prepare_image() (vissl.utils.instance_retrieval_utils.data_util.InstanceRetrievalImageLoader method) load_and_prepare_instre_image() (vissl.utils.instance_retrieval_utils.data_util.InstanceRetrievalImageLoader method) load_and_prepare_revisited_image() (vissl.utils.instance_retrieval_utils.data_util.InstanceRetrievalImageLoader method) load_and_prepare_whitening_image() (vissl.utils.instance_retrieval_utils.data_util.InstanceRetrievalImageLoader method)

load_file() (in module vissl.utils.io) load_input_data() (vissl.utils.svm_utils.svm_trainer.SVMTrainer method) load_pca() (in module vissl.utils.instance_retrieval_utils.pca) load_single_label_file() (vissl.data.GenericSSLDataset method) (vissl.data.ssl_dataset.GenericSSLDataset method) load_state_dict() (vissl.losses.moco_loss.MoCoLoss method) (vissl.losses.swav_momentum_loss.SwAVMomentumLoss method) log_gpu_stats() (in module vissl.utils.logger) LogGpuStatsHook (class in vissl.hooks.log_hooks) LogLossLrEtaHook (class in vissl.hooks.log_hooks) LogLossMetricsCheckpointHook (class in vissl.hooks.log_hooks) LogPerfTimeMetricsHook (class in vissl.hooks.log_hooks)

M

makedir() (in module vissl.utils.io) manual_gradient_all_reduce() (in module vissl.utils.activation_checkpointing) manual_gradient_reduction() (in module vissl.utils.activation_checkpointing) manual_sync_params() (in module vissl.utils.activation_checkpointing) MAX_PENDING_TIMERS (vissl.utils.perf_stats.PerfStats attribute) MeanAPListMeter (class in vissl.meters.mean_ap_list_meter) MeanAPMeter (class in vissl.meters.mean_ap_meter) merge_features() (in module vissl.utils.misc) MLP (class in vissl.models.heads) moco_collator() (in module vissl.data.collators.moco_collator) MoCoHook (class in vissl.hooks.moco_hooks) MoCoLoss (class in vissl.losses.moco_loss) MoCoLossConfig (class in vissl.losses.moco_loss) module vissl.data vissl.data.collators vissl.data.collators.mixup_collator vissl.data.collators.moco_collator vissl.data.collators.multicrop_collator vissl.data.collators.patch_and_image_collator vissl.data.collators.siamese_collator vissl.data.collators.simclr_collator vissl.data.collators.targets_one_hot_default_collator vissl.data.data_helper vissl.data.dataloader_sync_gpu_wrapper vissl.data.dataset_catalog vissl.data.disk_dataset vissl.data.ssl_dataset vissl.data.ssl_transforms vissl.data.ssl_transforms.img_patches_tensor vissl.data.ssl_transforms.img_pil_color_distortion vissl.data.ssl_transforms.img_pil_gaussian_blur vissl.data.ssl_transforms.img_pil_multicrop_random_apply vissl.data.ssl_transforms.img_pil_random_color_jitter vissl.data.ssl_transforms.img_pil_random_photometric vissl.data.ssl_transforms.img_pil_random_solarize vissl.data.ssl_transforms.img_pil_to_lab_tensor vissl.data.ssl_transforms.img_pil_to_multicrop vissl.data.ssl_transforms.img_pil_to_patches_and_image vissl.data.ssl_transforms.img_pil_to_raw_tensor vissl.data.ssl_transforms.img_pil_to_tensor vissl.data.ssl_transforms.img_replicate_pil vissl.data.ssl_transforms.img_rotate_pil vissl.data.ssl_transforms.pil_photometric_transforms_lib vissl.data.ssl_transforms.shuffle_img_patches vissl.data.synthetic_dataset vissl.engines.extract_features vissl.engines.train vissl.hooks vissl.hooks.deepclusterv2_hooks vissl.hooks.log_hooks vissl.hooks.moco_hooks vissl.hooks.state_update_hooks vissl.hooks.swav_hooks vissl.hooks.swav_momentum_hooks vissl.hooks.tensorboard_hook vissl.losses vissl.losses.bce_logits_multiple_output_single_target vissl.losses.cross_entropy_multiple_output_single_target vissl.losses.deepclusterv2_loss vissl.losses.moco_loss vissl.losses.multicrop_simclr_info_nce_loss vissl.losses.nce_loss vissl.losses.simclr_info_nce_loss vissl.losses.swav_loss vissl.losses.swav_momentum_loss vissl.meters.accuracy_list_meter vissl.meters.mean_ap_list_meter vissl.meters.mean_ap_meter vissl.models vissl.models.heads vissl.models.model_helpers vissl.models.trunks vissl.optimizers vissl.optimizers.optimizer_helper vissl.optimizers.param_scheduler.cosine_warm_restart_scheduler vissl.optimizers.param_scheduler.inverse_sqrt_decay vissl.trainer.train_steps vissl.trainer.train_steps.standard_train_step vissl.trainer.train_task vissl.trainer.trainer_main vissl.utils.activation_checkpointing vissl.utils.checkpoint vissl.utils.collect_env vissl.utils.env vissl.utils.hydra_config vissl.utils.instance_retrieval_utils.data_util vissl.utils.instance_retrieval_utils.evaluate vissl.utils.instance_retrieval_utils.pca vissl.utils.instance_retrieval_utils.rmac vissl.utils.io vissl.utils.logger vissl.utils.misc vissl.utils.perf_stats vissl.utils.slurm vissl.utils.svm_utils.evaluate vissl.utils.svm_utils.svm_low_shot_trainer vissl.utils.svm_utils.svm_trainer vissl.utils.tensorboard

multi_input_with_head_mapping_forward() (vissl.models.BaseSSLMultiInputOutputModel method) multi_res_input_forward() (vissl.models.BaseSSLMultiInputOutputModel method) multicrop_collator() (in module vissl.data.collators.multicrop_collator) multicrop_mixup_collator() (in module vissl.data.collators.mixup_collator) MultiCropSimclrInfoNCECriterion (class in vissl.losses.multicrop_simclr_info_nce_loss) MultiCropSimclrInfoNCELoss (class in vissl.losses.multicrop_simclr_info_nce_loss) MultigrainResize (class in vissl.utils.instance_retrieval_utils.data_util)

N

name() (vissl.meters.accuracy_list_meter.AccuracyListMeter property) (vissl.meters.mean_ap_list_meter.MeanAPListMeter property) (vissl.meters.mean_ap_meter.MeanAPMeter property) NCEAverage (class in vissl.losses.nce_loss) NCECriterion (class in vissl.losses.nce_loss) NCELossWithMemory (class in vissl.losses.nce_loss) normalize_L2() (in module vissl.utils.instance_retrieval_utils.rmac)

NormalizePrototypesHook (class in vissl.hooks.swav_hooks) num_classes() (vissl.models.BaseSSLMultiInputOutputModel property) num_samples() (vissl.data.disk_dataset.DiskImageDataset method) (vissl.data.GenericSSLDataset method) (vissl.data.ssl_dataset.GenericSSLDataset method) (vissl.data.synthetic_dataset.SyntheticImageDataset method) NumpySampler (class in vissl.losses.nce_loss)

O

on_backward (vissl.hooks.SSLClassyHookFunctions attribute) on_backward() (vissl.hooks.deepclusterv2_hooks.ClusterMemoryHook method) (vissl.hooks.deepclusterv2_hooks.InitMemoryHook method) (vissl.hooks.log_hooks.LogGpuStatsHook method) (vissl.hooks.log_hooks.LogLossLrEtaHook method) (vissl.hooks.log_hooks.LogLossMetricsCheckpointHook method) (vissl.hooks.log_hooks.LogPerfTimeMetricsHook method) (vissl.hooks.moco_hooks.MoCoHook method) (vissl.hooks.state_update_hooks.CheckNanLossHook method) (vissl.hooks.state_update_hooks.FreezeParametersHook method) (vissl.hooks.state_update_hooks.SetDataSamplerEpochHook method) (vissl.hooks.state_update_hooks.SSLModelComplexityHook method) (vissl.hooks.state_update_hooks.UpdateBatchesSeenHook method) (vissl.hooks.state_update_hooks.UpdateTestBatchTimeHook method) (vissl.hooks.state_update_hooks.UpdateTrainBatchTimeHook method) (vissl.hooks.state_update_hooks.UpdateTrainIterationNumHook method) (vissl.hooks.swav_hooks.NormalizePrototypesHook method) (vissl.hooks.swav_hooks.SwAVUpdateQueueScoresHook method) (vissl.hooks.swav_momentum_hooks.SwAVMomentumHook method) (vissl.hooks.swav_momentum_hooks.SwAVMomentumNormalizePrototypesHook method) (vissl.hooks.tensorboard_hook.SSLTensorboardHook method) on_end (vissl.hooks.SSLClassyHookFunctions attribute) on_end() (vissl.hooks.deepclusterv2_hooks.ClusterMemoryHook method) (vissl.hooks.deepclusterv2_hooks.InitMemoryHook method) (vissl.hooks.log_hooks.LogGpuStatsHook method) (vissl.hooks.log_hooks.LogLossLrEtaHook method) (vissl.hooks.log_hooks.LogLossMetricsCheckpointHook method) (vissl.hooks.log_hooks.LogPerfTimeMetricsHook method) (vissl.hooks.moco_hooks.MoCoHook method) (vissl.hooks.state_update_hooks.CheckNanLossHook method) (vissl.hooks.state_update_hooks.FreezeParametersHook method) (vissl.hooks.state_update_hooks.SetDataSamplerEpochHook method) (vissl.hooks.state_update_hooks.SSLModelComplexityHook method) (vissl.hooks.state_update_hooks.UpdateBatchesSeenHook method) (vissl.hooks.state_update_hooks.UpdateTestBatchTimeHook method) (vissl.hooks.state_update_hooks.UpdateTrainBatchTimeHook method) (vissl.hooks.state_update_hooks.UpdateTrainIterationNumHook method) (vissl.hooks.swav_hooks.NormalizePrototypesHook method) (vissl.hooks.swav_hooks.SwAVUpdateQueueScoresHook method) (vissl.hooks.swav_momentum_hooks.SwAVMomentumHook method) (vissl.hooks.swav_momentum_hooks.SwAVMomentumNormalizePrototypesHook method) (vissl.hooks.tensorboard_hook.SSLTensorboardHook method) on_failure() (vissl.data.data_helper.QueueDataset method) on_forward (vissl.hooks.SSLClassyHookFunctions attribute) on_forward() (vissl.hooks.deepclusterv2_hooks.ClusterMemoryHook method) (vissl.hooks.deepclusterv2_hooks.InitMemoryHook method) (vissl.hooks.log_hooks.LogGpuStatsHook method) (vissl.hooks.log_hooks.LogLossLrEtaHook method) (vissl.hooks.log_hooks.LogLossMetricsCheckpointHook method) (vissl.hooks.log_hooks.LogPerfTimeMetricsHook method) (vissl.hooks.moco_hooks.MoCoHook method) (vissl.hooks.state_update_hooks.CheckNanLossHook method) (vissl.hooks.state_update_hooks.FreezeParametersHook method) (vissl.hooks.state_update_hooks.SetDataSamplerEpochHook method) (vissl.hooks.state_update_hooks.SSLModelComplexityHook method) (vissl.hooks.state_update_hooks.UpdateBatchesSeenHook method) (vissl.hooks.state_update_hooks.UpdateTestBatchTimeHook method) (vissl.hooks.state_update_hooks.UpdateTrainBatchTimeHook method) (vissl.hooks.state_update_hooks.UpdateTrainIterationNumHook method) (vissl.hooks.swav_hooks.NormalizePrototypesHook method) (vissl.hooks.swav_hooks.SwAVUpdateQueueScoresHook method) (vissl.hooks.swav_momentum_hooks.SwAVMomentumHook method) (vissl.hooks.swav_momentum_hooks.SwAVMomentumNormalizePrototypesHook method) (vissl.hooks.tensorboard_hook.SSLTensorboardHook method) on_loss_and_meter (vissl.hooks.SSLClassyHookFunctions attribute) on_loss_and_meter() (vissl.hooks.deepclusterv2_hooks.ClusterMemoryHook method) (vissl.hooks.deepclusterv2_hooks.InitMemoryHook method) (vissl.hooks.log_hooks.LogGpuStatsHook method) (vissl.hooks.log_hooks.LogLossLrEtaHook method) (vissl.hooks.log_hooks.LogLossMetricsCheckpointHook method) (vissl.hooks.log_hooks.LogPerfTimeMetricsHook method) (vissl.hooks.moco_hooks.MoCoHook method) (vissl.hooks.state_update_hooks.CheckNanLossHook method) (vissl.hooks.state_update_hooks.FreezeParametersHook method) (vissl.hooks.state_update_hooks.SetDataSamplerEpochHook method) (vissl.hooks.state_update_hooks.SSLModelComplexityHook method) (vissl.hooks.state_update_hooks.UpdateBatchesSeenHook method) (vissl.hooks.state_update_hooks.UpdateTestBatchTimeHook method) (vissl.hooks.state_update_hooks.UpdateTrainBatchTimeHook method) (vissl.hooks.state_update_hooks.UpdateTrainIterationNumHook method) (vissl.hooks.swav_hooks.NormalizePrototypesHook method) (vissl.hooks.swav_hooks.SwAVUpdateQueueScoresHook method) (vissl.hooks.swav_momentum_hooks.SwAVMomentumHook method) (vissl.hooks.swav_momentum_hooks.SwAVMomentumNormalizePrototypesHook method) (vissl.hooks.tensorboard_hook.SSLTensorboardHook method) on_phase_end (vissl.hooks.SSLClassyHookFunctions attribute) on_phase_end() (vissl.hooks.deepclusterv2_hooks.ClusterMemoryHook method) (vissl.hooks.deepclusterv2_hooks.InitMemoryHook method) (vissl.hooks.log_hooks.LogGpuStatsHook method) (vissl.hooks.log_hooks.LogLossLrEtaHook method) (vissl.hooks.log_hooks.LogLossMetricsCheckpointHook method) (vissl.hooks.log_hooks.LogPerfTimeMetricsHook method) (vissl.hooks.moco_hooks.MoCoHook method) (vissl.hooks.state_update_hooks.CheckNanLossHook method) (vissl.hooks.state_update_hooks.FreezeParametersHook method) (vissl.hooks.state_update_hooks.SetDataSamplerEpochHook method) (vissl.hooks.state_update_hooks.SSLModelComplexityHook method) (vissl.hooks.state_update_hooks.UpdateBatchesSeenHook method) (vissl.hooks.state_update_hooks.UpdateTestBatchTimeHook method) (vissl.hooks.state_update_hooks.UpdateTrainBatchTimeHook method) (vissl.hooks.state_update_hooks.UpdateTrainIterationNumHook method) (vissl.hooks.swav_hooks.NormalizePrototypesHook method) (vissl.hooks.swav_hooks.SwAVUpdateQueueScoresHook method) (vissl.hooks.swav_momentum_hooks.SwAVMomentumHook method) (vissl.hooks.swav_momentum_hooks.SwAVMomentumNormalizePrototypesHook method) (vissl.hooks.tensorboard_hook.SSLTensorboardHook method)

on_phase_start (vissl.hooks.SSLClassyHookFunctions attribute) on_phase_start() (vissl.hooks.deepclusterv2_hooks.ClusterMemoryHook method) (vissl.hooks.deepclusterv2_hooks.InitMemoryHook method) (vissl.hooks.log_hooks.LogGpuStatsHook method) (vissl.hooks.log_hooks.LogLossLrEtaHook method) (vissl.hooks.log_hooks.LogLossMetricsCheckpointHook method) (vissl.hooks.log_hooks.LogPerfTimeMetricsHook method) (vissl.hooks.moco_hooks.MoCoHook method) (vissl.hooks.state_update_hooks.CheckNanLossHook method) (vissl.hooks.state_update_hooks.FreezeParametersHook method) (vissl.hooks.state_update_hooks.SetDataSamplerEpochHook method) (vissl.hooks.state_update_hooks.SSLModelComplexityHook method) (vissl.hooks.state_update_hooks.UpdateBatchesSeenHook method) (vissl.hooks.state_update_hooks.UpdateTestBatchTimeHook method) (vissl.hooks.state_update_hooks.UpdateTrainBatchTimeHook method) (vissl.hooks.state_update_hooks.UpdateTrainIterationNumHook method) (vissl.hooks.swav_hooks.NormalizePrototypesHook method) (vissl.hooks.swav_hooks.SwAVUpdateQueueScoresHook method) (vissl.hooks.swav_momentum_hooks.SwAVMomentumHook method) (vissl.hooks.swav_momentum_hooks.SwAVMomentumNormalizePrototypesHook method) (vissl.hooks.tensorboard_hook.SSLTensorboardHook method) on_start (vissl.hooks.SSLClassyHookFunctions attribute) on_start() (vissl.hooks.deepclusterv2_hooks.ClusterMemoryHook method) (vissl.hooks.deepclusterv2_hooks.InitMemoryHook method) (vissl.hooks.log_hooks.LogGpuStatsHook method) (vissl.hooks.log_hooks.LogLossLrEtaHook method) (vissl.hooks.log_hooks.LogLossMetricsCheckpointHook method) (vissl.hooks.log_hooks.LogPerfTimeMetricsHook method) (vissl.hooks.moco_hooks.MoCoHook method) (vissl.hooks.state_update_hooks.CheckNanLossHook method) (vissl.hooks.state_update_hooks.FreezeParametersHook method) (vissl.hooks.state_update_hooks.SetDataSamplerEpochHook method) (vissl.hooks.state_update_hooks.SSLModelComplexityHook method) (vissl.hooks.state_update_hooks.UpdateBatchesSeenHook method) (vissl.hooks.state_update_hooks.UpdateTestBatchTimeHook method) (vissl.hooks.state_update_hooks.UpdateTrainBatchTimeHook method) (vissl.hooks.state_update_hooks.UpdateTrainIterationNumHook method) (vissl.hooks.swav_hooks.NormalizePrototypesHook method) (vissl.hooks.swav_hooks.SwAVUpdateQueueScoresHook method) (vissl.hooks.swav_momentum_hooks.SwAVMomentumHook method) (vissl.hooks.swav_momentum_hooks.SwAVMomentumNormalizePrototypesHook method) (vissl.hooks.tensorboard_hook.SSLTensorboardHook method) on_step (vissl.hooks.SSLClassyHookFunctions attribute) on_step() (vissl.hooks.deepclusterv2_hooks.ClusterMemoryHook method) (vissl.hooks.deepclusterv2_hooks.InitMemoryHook method) (vissl.hooks.log_hooks.LogGpuStatsHook method) (vissl.hooks.log_hooks.LogLossLrEtaHook method) (vissl.hooks.log_hooks.LogLossMetricsCheckpointHook method) (vissl.hooks.log_hooks.LogPerfTimeMetricsHook method) (vissl.hooks.moco_hooks.MoCoHook method) (vissl.hooks.state_update_hooks.CheckNanLossHook method) (vissl.hooks.state_update_hooks.FreezeParametersHook method) (vissl.hooks.state_update_hooks.SetDataSamplerEpochHook method) (vissl.hooks.state_update_hooks.SSLModelComplexityHook method) (vissl.hooks.state_update_hooks.UpdateBatchesSeenHook method) (vissl.hooks.state_update_hooks.UpdateTestBatchTimeHook method) (vissl.hooks.state_update_hooks.UpdateTrainBatchTimeHook method) (vissl.hooks.state_update_hooks.UpdateTrainIterationNumHook method) (vissl.hooks.swav_hooks.NormalizePrototypesHook method) (vissl.hooks.swav_hooks.SwAVUpdateQueueScoresHook method) (vissl.hooks.swav_momentum_hooks.SwAVMomentumHook method) (vissl.hooks.swav_momentum_hooks.SwAVMomentumNormalizePrototypesHook method) (vissl.hooks.tensorboard_hook.SSLTensorboardHook method) on_sucess() (vissl.data.data_helper.QueueDataset method) on_update (vissl.hooks.SSLClassyHookFunctions attribute) on_update() (vissl.hooks.deepclusterv2_hooks.ClusterMemoryHook method) (vissl.hooks.deepclusterv2_hooks.InitMemoryHook method) (vissl.hooks.log_hooks.LogGpuStatsHook method) (vissl.hooks.log_hooks.LogLossLrEtaHook method) (vissl.hooks.log_hooks.LogLossMetricsCheckpointHook method) (vissl.hooks.log_hooks.LogPerfTimeMetricsHook method) (vissl.hooks.moco_hooks.MoCoHook method) (vissl.hooks.state_update_hooks.CheckNanLossHook method) (vissl.hooks.state_update_hooks.FreezeParametersHook method) (vissl.hooks.state_update_hooks.SetDataSamplerEpochHook method) (vissl.hooks.state_update_hooks.SSLModelComplexityHook method) (vissl.hooks.state_update_hooks.UpdateBatchesSeenHook method) (vissl.hooks.state_update_hooks.UpdateTestBatchTimeHook method) (vissl.hooks.state_update_hooks.UpdateTrainBatchTimeHook method) (vissl.hooks.state_update_hooks.UpdateTrainIterationNumHook method) (vissl.hooks.swav_hooks.NormalizePrototypesHook method) (vissl.hooks.swav_hooks.SwAVUpdateQueueScoresHook method) (vissl.hooks.swav_momentum_hooks.SwAVMomentumHook method) (vissl.hooks.swav_momentum_hooks.SwAVMomentumNormalizePrototypesHook method) (vissl.hooks.tensorboard_hook.SSLTensorboardHook method) output_shape() (vissl.models.BaseSSLMultiInputOutputModel property)

P

parse_out_keys_arg() (in module vissl.models.model_helpers) patch_and_image_collator() (in module vissl.data.collators.patch_and_image_collator) PCA (class in vissl.utils.instance_retrieval_utils.pca) PerfMetric (class in vissl.utils.perf_stats) PerfStats (class in vissl.utils.perf_stats) PerfTimer (class in vissl.utils.perf_stats) Posterize() (in module vissl.data.ssl_transforms.pil_photometric_transforms_lib) precompute_pos_neg_mask() (vissl.losses.multicrop_simclr_info_nce_loss.MultiCropSimclrInfoNCECriterion method) (vissl.losses.simclr_info_nce_loss.SimclrInfoNCECriterion method)

prepare() (vissl.trainer.train_task.SelfSupervisionTask method) prepare_extraction() (vissl.trainer.train_task.SelfSupervisionTask method) prepare_optimizer() (vissl.trainer.train_task.SelfSupervisionTask method) print_cfg() (in module vissl.utils.hydra_config) print_gpu_memory_usage() (in module vissl.utils.logger) print_loaded_dict_info() (in module vissl.utils.checkpoint) print_state_dict_shapes() (in module vissl.utils.checkpoint) print_system_env_info() (in module vissl.utils.env) pytorch (vissl.models.model_helpers.SyncBNTypes attribute)

Q

QueueDataset (class in vissl.data.data_helper)

R

RandomPosterizeTransform (class in vissl.data.ssl_transforms.pil_photometric_transforms_lib) RandomSharpnessTransform (class in vissl.data.ssl_transforms.pil_photometric_transforms_lib) RandomSolarizeTransform (class in vissl.data.ssl_transforms.pil_photometric_transforms_lib) RandomValueApplier (class in vissl.data.ssl_transforms.pil_photometric_transforms_lib) record() (vissl.utils.perf_stats.PerfTimer method) recreate_data_iterator() (vissl.trainer.train_task.SelfSupervisionTask method) register_coco() (in module vissl.data.dataset_catalog) register_collator() (in module vissl.data.collators) register_data() (vissl.data.dataset_catalog.VisslDatasetCatalog static method) (vissl.data.VisslDatasetCatalog static method) register_datasets() (in module vissl.data) (in module vissl.data.dataset_catalog) register_dict() (vissl.data.dataset_catalog.VisslDatasetCatalog static method) (vissl.data.VisslDatasetCatalog static method) register_json() (vissl.data.dataset_catalog.VisslDatasetCatalog static method) (vissl.data.VisslDatasetCatalog static method)

register_model_trunk() (in module vissl.models.trunks) register_pascal_voc() (in module vissl.data.dataset_catalog) register_train_step() (in module vissl.trainer.train_steps) remove() (vissl.data.dataset_catalog.VisslDatasetCatalog static method) (vissl.data.VisslDatasetCatalog static method) replace_module_prefix() (in module vissl.utils.checkpoint) report_str() (vissl.utils.perf_stats.PerfStats method) reset() (vissl.meters.accuracy_list_meter.AccuracyListMeter method) (vissl.meters.mean_ap_list_meter.MeanAPListMeter method) (vissl.meters.mean_ap_meter.MeanAPMeter method) RESNET_NORM_LAYER (class in vissl.models.model_helpers) resolve_linear_schedule() (in module vissl.utils.hydra_config) RevisitedInstanceRetrievalDataset (class in vissl.utils.instance_retrieval_utils.data_util) run_hooks() (vissl.trainer.train_task.SelfSupervisionTask method)

S

sample_value() (vissl.data.ssl_transforms.pil_photometric_transforms_lib.RandomValueApplier method) save_file() (in module vissl.utils.io) scale_weights() (vissl.models.heads.MLP method) score() (vissl.utils.instance_retrieval_utils.data_util.InstanceRetrievalDataset method) (vissl.utils.instance_retrieval_utils.data_util.InstreDataset method) (vissl.utils.instance_retrieval_utils.data_util.RevisitedInstanceRetrievalDataset method) score_ap_from_ranks_1() (in module vissl.utils.instance_retrieval_utils.evaluate) score_rnk_partial() (vissl.utils.instance_retrieval_utils.data_util.InstanceRetrievalDataset method) SelfSupervisionTask (class in vissl.trainer.train_task) SelfSupervisionTrainer (class in vissl.trainer.trainer_main) set_amp_args() (vissl.trainer.train_task.SelfSupervisionTask method) set_available_splits() (vissl.trainer.train_task.SelfSupervisionTask method) set_checkpoint_folder() (vissl.trainer.train_task.SelfSupervisionTask method) set_checkpoint_path() (vissl.trainer.train_task.SelfSupervisionTask method) set_classy_state() (vissl.meters.accuracy_list_meter.AccuracyListMeter method) (vissl.meters.mean_ap_list_meter.MeanAPListMeter method) (vissl.meters.mean_ap_meter.MeanAPMeter method) (vissl.models.BaseSSLMultiInputOutputModel method) set_ddp_bucket_cap_mb() (vissl.trainer.train_task.SelfSupervisionTask method) set_device() (vissl.trainer.train_task.SelfSupervisionTask method) set_env_vars() (in module vissl.utils.env) set_iteration() (vissl.trainer.train_task.SelfSupervisionTask method) set_manual_gradient_reduction() (vissl.trainer.train_task.SelfSupervisionTask method) set_seeds() (in module vissl.utils.misc) set_start_iter() (vissl.data.data_helper.StatefulDistributedSampler method) SetDataSamplerEpochHook (class in vissl.hooks.state_update_hooks) setup_distributed() (vissl.trainer.trainer_main.SelfSupervisionTrainer method) setup_logging() (in module vissl.utils.logger) setup_multiprocessing_method() (in module vissl.utils.misc) setup_negative_sampling() (vissl.losses.nce_loss.NCEAverage method) Sharpness() (in module vissl.data.ssl_transforms.pil_photometric_transforms_lib)

ShuffleImgPatches (class in vissl.data.ssl_transforms.shuffle_img_patches) shutdown_logging() (in module vissl.utils.logger) siamese_collator() (in module vissl.data.collators.siamese_collator) SiameseConcatView (class in vissl.models.heads) simclr_collator() (in module vissl.data.collators.simclr_collator) SimclrInfoNCECriterion (class in vissl.losses.simclr_info_nce_loss) SimclrInfoNCELoss (class in vissl.losses.simclr_info_nce_loss) single_input_forward() (vissl.models.BaseSSLMultiInputOutputModel method) Solarize() (in module vissl.data.ssl_transforms.pil_photometric_transforms_lib) SSLClassyHookFunctions (class in vissl.hooks) SSLModelComplexityHook (class in vissl.hooks.state_update_hooks) SSLTensorboardHook (class in vissl.hooks.tensorboard_hook) SSLTransformsWrapper (class in vissl.data.ssl_transforms) standard_train_step() (in module vissl.trainer.train_steps.standard_train_step) start() (vissl.utils.perf_stats.PerfTimer method) StatefulDistributedSampler (class in vissl.data.data_helper) stop() (vissl.utils.perf_stats.PerfTimer method) SVMLowShotTrainer (class in vissl.utils.svm_utils.svm_low_shot_trainer) SVMTrainer (class in vissl.utils.svm_utils.svm_trainer) SwAVCriterion (class in vissl.losses.swav_loss) SwAVLoss (class in vissl.losses.swav_loss) SwAVMomentumHook (class in vissl.hooks.swav_momentum_hooks) SwAVMomentumLoss (class in vissl.losses.swav_momentum_loss) SwAVMomentumNormalizePrototypesHook (class in vissl.hooks.swav_momentum_hooks) SwAVPrototypesHead (class in vissl.models.heads) SwAVUpdateQueueScoresHook (class in vissl.hooks.swav_hooks) sync_memory() (vissl.losses.nce_loss.NCELossWithMemory method) sync_state() (vissl.meters.accuracy_list_meter.AccuracyListMeter method) (vissl.meters.mean_ap_list_meter.MeanAPListMeter method) (vissl.meters.mean_ap_meter.MeanAPMeter method) SyncBNTypes (class in vissl.models.model_helpers) SyntheticImageDataset (class in vissl.data.synthetic_dataset)

T

target_size() (vissl.utils.instance_retrieval_utils.data_util.MultigrainResize static method) targets_one_hot_default_collator() (in module vissl.data.collators.targets_one_hot_default_collator) test() (vissl.utils.svm_utils.svm_low_shot_trainer.SVMLowShotTrainer method) (vissl.utils.svm_utils.svm_trainer.SVMTrainer method) to_cuda() (vissl.utils.instance_retrieval_utils.pca.PCA method) train() (vissl.trainer.trainer_main.SelfSupervisionTrainer method) (vissl.utils.svm_utils.svm_low_shot_trainer.SVMLowShotTrainer method) (vissl.utils.svm_utils.svm_trainer.SVMTrainer method)

train_and_save_pca() (in module vissl.utils.instance_retrieval_utils.pca) train_cls() (vissl.utils.svm_utils.svm_trainer.SVMTrainer method) train_main() (in module vissl.engines.train) transform_model_input_data_type() (in module vissl.models.model_helpers) TransformObject (class in vissl.data.ssl_transforms.pil_photometric_transforms_lib)

U

update() (vissl.meters.accuracy_list_meter.AccuracyListMeter method) (vissl.meters.mean_ap_list_meter.MeanAPListMeter method) (vissl.meters.mean_ap_meter.MeanAPMeter method) (vissl.utils.perf_stats.PerfMetric method) update_emb_queue() (vissl.losses.swav_loss.SwAVCriterion method) (vissl.losses.swav_momentum_loss.SwAVMomentumLoss method) update_memory() (vissl.losses.nce_loss.NCEAverage method) (vissl.losses.nce_loss.NCELossWithMemory method)

update_memory_bank() (vissl.losses.deepclusterv2_loss.DeepClusterV2Loss method) update_with_timer() (vissl.utils.perf_stats.PerfStats method) UpdateBatchesSeenHook (class in vissl.hooks.state_update_hooks) UpdateTestBatchTimeHook (class in vissl.hooks.state_update_hooks) UpdateTrainBatchTimeHook (class in vissl.hooks.state_update_hooks) UpdateTrainIterationNumHook (class in vissl.hooks.state_update_hooks) use_cuda_events() (vissl.utils.perf_stats.PerfStats method)

V

validate() (vissl.meters.accuracy_list_meter.AccuracyListMeter method) (vissl.meters.mean_ap_list_meter.MeanAPListMeter method) (vissl.meters.mean_ap_meter.MeanAPMeter method) (vissl.models.BaseSSLMultiInputOutputModel method) value() (vissl.meters.accuracy_list_meter.AccuracyListMeter property) (vissl.meters.mean_ap_list_meter.MeanAPListMeter property) (vissl.meters.mean_ap_meter.MeanAPMeter property) verify_target() (vissl.meters.mean_ap_meter.MeanAPMeter method) vissl.data module vissl.data.collators module vissl.data.collators.mixup_collator module vissl.data.collators.moco_collator module vissl.data.collators.multicrop_collator module vissl.data.collators.patch_and_image_collator module vissl.data.collators.siamese_collator module vissl.data.collators.simclr_collator module vissl.data.collators.targets_one_hot_default_collator module vissl.data.data_helper module vissl.data.dataloader_sync_gpu_wrapper module vissl.data.dataset_catalog module vissl.data.disk_dataset module vissl.data.ssl_dataset module vissl.data.ssl_transforms module vissl.data.ssl_transforms.img_patches_tensor module vissl.data.ssl_transforms.img_pil_color_distortion module vissl.data.ssl_transforms.img_pil_gaussian_blur module vissl.data.ssl_transforms.img_pil_multicrop_random_apply module vissl.data.ssl_transforms.img_pil_random_color_jitter module vissl.data.ssl_transforms.img_pil_random_photometric module vissl.data.ssl_transforms.img_pil_random_solarize module vissl.data.ssl_transforms.img_pil_to_lab_tensor module vissl.data.ssl_transforms.img_pil_to_multicrop module vissl.data.ssl_transforms.img_pil_to_patches_and_image module vissl.data.ssl_transforms.img_pil_to_raw_tensor module vissl.data.ssl_transforms.img_pil_to_tensor module vissl.data.ssl_transforms.img_replicate_pil module vissl.data.ssl_transforms.img_rotate_pil module vissl.data.ssl_transforms.pil_photometric_transforms_lib module vissl.data.ssl_transforms.shuffle_img_patches module vissl.data.synthetic_dataset module vissl.engines.extract_features module vissl.engines.train module vissl.hooks module vissl.hooks.deepclusterv2_hooks module vissl.hooks.log_hooks module vissl.hooks.moco_hooks module vissl.hooks.state_update_hooks module vissl.hooks.swav_hooks module vissl.hooks.swav_momentum_hooks module

vissl.hooks.tensorboard_hook module vissl.losses module vissl.losses.bce_logits_multiple_output_single_target module vissl.losses.cross_entropy_multiple_output_single_target module vissl.losses.deepclusterv2_loss module vissl.losses.moco_loss module vissl.losses.multicrop_simclr_info_nce_loss module vissl.losses.nce_loss module vissl.losses.simclr_info_nce_loss module vissl.losses.swav_loss module vissl.losses.swav_momentum_loss module vissl.meters.accuracy_list_meter module vissl.meters.mean_ap_list_meter module vissl.meters.mean_ap_meter module vissl.models module vissl.models.heads module vissl.models.model_helpers module vissl.models.trunks module vissl.optimizers module vissl.optimizers.optimizer_helper module vissl.optimizers.param_scheduler.cosine_warm_restart_scheduler module vissl.optimizers.param_scheduler.inverse_sqrt_decay module vissl.trainer.train_steps module vissl.trainer.train_steps.standard_train_step module vissl.trainer.train_task module vissl.trainer.trainer_main module vissl.utils.activation_checkpointing module vissl.utils.checkpoint module vissl.utils.collect_env module vissl.utils.env module vissl.utils.hydra_config module vissl.utils.instance_retrieval_utils.data_util module vissl.utils.instance_retrieval_utils.evaluate module vissl.utils.instance_retrieval_utils.pca module vissl.utils.instance_retrieval_utils.rmac module vissl.utils.io module vissl.utils.logger module vissl.utils.misc module vissl.utils.perf_stats module vissl.utils.slurm module vissl.utils.svm_utils.evaluate module vissl.utils.svm_utils.svm_low_shot_trainer module vissl.utils.svm_utils.svm_trainer module vissl.utils.tensorboard module VisslDatasetCatalog (class in vissl.data) (class in vissl.data.dataset_catalog)

W

WhiteningTrainingImageDataset (class in vissl.utils.instance_retrieval_utils.data_util)

Wrap (class in vissl.models.model_helpers)