Index

_

__init__() (quanda.benchmarks.resources.modules.BertClassifier method) (quanda.benchmarks.resources.modules.HFGPT2 method) (quanda.benchmarks.resources.modules.LeNet method) (quanda.benchmarks.resources.modules.ResNet50 method) (quanda.benchmarks.resources.modules.ResNet9 method) (quanda.utils.cache.BatchedCachedExplanations method) (quanda.utils.common.DatasetSplit method) (quanda.utils.datasets.dataset_handlers.HuggingFaceSequenceDatasetHandler method) (quanda.utils.datasets.image_datasets.HFtoTV method) (quanda.utils.datasets.on_device_dataset.OnDeviceDataset method) (quanda.utils.datasets.OnDeviceDataset method) (quanda.utils.training.base_pl_module.BasicLightningModule method) (quanda.utils.training.BasicLightningModule method) (quanda.utils.training.Trainer method) (quanda.utils.training.trainer.Trainer method)

A

add_pink_frame() (in module quanda.benchmarks.resources.sample_transforms)

add_white_square_mnist() (in module quanda.benchmarks.resources.sample_transforms) add_yellow_square() (in module quanda.benchmarks.resources.sample_transforms)

B

BaseTrainer (class in quanda.utils.training) (class in quanda.utils.training.trainer) BasicLightningModule (class in quanda.utils.training) (class in quanda.utils.training.base_pl_module)

BatchedCachedExplanations (class in quanda.utils.cache) BertClassifier (class in quanda.benchmarks.resources.modules) build_positional_batch() (quanda.utils.datasets.dataset_handlers.HuggingFaceSequenceDatasetHandler method) (quanda.utils.datasets.dataset_handlers.TorchDatasetHandler method)

C

Cache (class in quanda.utils.cache) cache_result() (in module quanda.utils.common) chunked_logits() (in module quanda.utils.common) class_accuracy() (in module quanda.utils.common) collate() (quanda.utils.datasets.dataset_handlers.HuggingFaceSequenceDatasetHandler method) config_class (quanda.benchmarks.resources.modules.HFGPT2 attribute) configure_optimizers() (quanda.utils.training.base_pl_module.BasicLightningModule method) (quanda.utils.training.BasicLightningModule method)

cosine_similarity() (in module quanda.utils.functions) (in module quanda.utils.functions.similarities) create_dataloader() (quanda.utils.datasets.dataset_handlers.DatasetHandler method) (quanda.utils.datasets.dataset_handlers.HuggingFaceDatasetHandler method) (quanda.utils.datasets.dataset_handlers.HuggingFaceSequenceDatasetHandler method) (quanda.utils.datasets.dataset_handlers.TorchDatasetHandler method)

D

DatasetHandler (class in quanda.utils.datasets.dataset_handlers) DatasetSplit (class in quanda.utils.common) default_tensor_type() (in module quanda.utils.common)

dot_product_similarity() (in module quanda.utils.functions) (in module quanda.utils.functions.similarities) ds_len() (in module quanda.utils.common)

E

exists() (quanda.utils.cache.Cache static method) (quanda.utils.cache.ExplanationsCache static method) (quanda.utils.common.DatasetSplit static method)

ExplanationsCache (class in quanda.utils.cache)

F

fit() (quanda.utils.training.BaseTrainer method) (quanda.utils.training.Trainer method) (quanda.utils.training.trainer.BaseTrainer method) (quanda.utils.training.trainer.Trainer method) forward() (quanda.benchmarks.resources.modules.BertClassifier method) (quanda.benchmarks.resources.modules.HFGPT2 method) (quanda.benchmarks.resources.modules.LeNet method) (quanda.benchmarks.resources.modules.ResNet9 method) (quanda.utils.training.base_pl_module.BasicLightningModule method) (quanda.utils.training.BasicLightningModule method)

from_pretrained_base() (quanda.benchmarks.resources.modules.BertClassifier class method)

G

get_dataset_handler() (in module quanda.utils.datasets.dataset_handlers) get_label() (quanda.utils.datasets.dataset_handlers.DatasetHandler method) (quanda.utils.datasets.dataset_handlers.HuggingFaceDatasetHandler method) (quanda.utils.datasets.dataset_handlers.TorchDatasetHandler method) (quanda.utils.datasets.image_datasets.HFtoTV method) get_load_state_dict_func() (in module quanda.utils.common) get_model() (quanda.utils.training.BaseTrainer method) (quanda.utils.training.trainer.BaseTrainer method)

get_model_inputs() (quanda.utils.datasets.dataset_handlers.DatasetHandler method) (quanda.utils.datasets.dataset_handlers.HuggingFaceDatasetHandler method) (quanda.utils.datasets.dataset_handlers.TorchDatasetHandler method) get_parent_module_from_name() (in module quanda.utils.common) get_predictions() (quanda.utils.datasets.dataset_handlers.DatasetHandler method) (quanda.utils.datasets.dataset_handlers.HuggingFaceDatasetHandler method) (quanda.utils.datasets.dataset_handlers.TorchDatasetHandler method) get_targets() (in module quanda.utils.common)

H

HFGPT2 (class in quanda.benchmarks.resources.modules) HFtoTV (class in quanda.utils.datasets.image_datasets)

HuggingFaceDatasetHandler (class in quanda.utils.datasets.dataset_handlers) HuggingFaceSequenceDatasetHandler (class in quanda.utils.datasets.dataset_handlers)

K

kendall_rank_corr() (in module quanda.utils.functions) (in module quanda.utils.functions.correlations)

keys() (quanda.utils.cache.BatchedCachedExplanations method)

L

LeNet (class in quanda.benchmarks.resources.modules) load() (quanda.utils.cache.Cache static method) (quanda.utils.cache.ExplanationsCache static method) (quanda.utils.common.DatasetSplit class method)

load_last_checkpoint() (in module quanda.utils.common)

M

make_func() (in module quanda.utils.common) map_location_context() (in module quanda.utils.common) module quanda.benchmarks.resources quanda.benchmarks.resources.config_map quanda.benchmarks.resources.modules quanda.benchmarks.resources.sample_transforms quanda.utils quanda.utils.cache quanda.utils.common quanda.utils.datasets quanda.utils.datasets.dataset_handlers quanda.utils.datasets.image_datasets quanda.utils.datasets.on_device_dataset quanda.utils.functions quanda.utils.functions.correlations quanda.utils.functions.similarities quanda.utils.tasks quanda.utils.tokenization quanda.utils.training quanda.utils.training.base_pl_module quanda.utils.training.options quanda.utils.training.trainer

move_ds_item_to_device() (in module quanda.utils.common)

O

on_load_checkpoint() (quanda.utils.training.base_pl_module.BasicLightningModule method) (quanda.utils.training.BasicLightningModule method) on_save_checkpoint() (quanda.utils.training.base_pl_module.BasicLightningModule method) (quanda.utils.training.BasicLightningModule method)

OnDeviceDataset (class in quanda.utils.datasets) (class in quanda.utils.datasets.on_device_dataset)

P

process_batch() (quanda.utils.datasets.dataset_handlers.DatasetHandler method) (quanda.utils.datasets.dataset_handlers.HuggingFaceDatasetHandler method) (quanda.utils.datasets.dataset_handlers.HuggingFaceSequenceDatasetHandler method) (quanda.utils.datasets.dataset_handlers.TorchDatasetHandler method)

process_targets() (in module quanda.utils.common)

Q

quanda.benchmarks.resources module quanda.benchmarks.resources.config_map module quanda.benchmarks.resources.modules module quanda.benchmarks.resources.sample_transforms module quanda.utils module quanda.utils.cache module quanda.utils.common module quanda.utils.datasets module quanda.utils.datasets.dataset_handlers module quanda.utils.datasets.image_datasets module

quanda.utils.datasets.on_device_dataset module quanda.utils.functions module quanda.utils.functions.correlations module quanda.utils.functions.similarities module quanda.utils.tasks module quanda.utils.tokenization module quanda.utils.training module quanda.utils.training.base_pl_module module quanda.utils.training.options module quanda.utils.training.trainer module

R

replace_conv1d_with_linear() (in module quanda.utils.common) ResNet50 (class in quanda.benchmarks.resources.modules) ResNet9 (class in quanda.benchmarks.resources.modules)

resolve_config() (in module quanda.utils.common) resolve_device() (in module quanda.utils.common) resolve_tokenizer() (in module quanda.utils.tokenization)

S

save() (quanda.utils.cache.Cache static method) (quanda.utils.cache.ExplanationsCache static method) (quanda.utils.common.DatasetSplit method) spearman_rank_corr() (in module quanda.utils.functions) (in module quanda.utils.functions.correlations)

split() (quanda.utils.common.DatasetSplit class method) splits (quanda.utils.common.DatasetSplit attribute) stable_repr() (in module quanda.utils.common) subsample_dataset() (in module quanda.utils.common) subsample_indices() (in module quanda.utils.common)

T

to_dict() (quanda.utils.common.DatasetSplit method) tokenize_dataset() (in module quanda.utils.tokenization) TorchDatasetHandler (class in quanda.utils.datasets.dataset_handlers)

Trainer (class in quanda.utils.training) (class in quanda.utils.training.trainer) training_step() (quanda.utils.training.base_pl_module.BasicLightningModule method) (quanda.utils.training.BasicLightningModule method)

V

validation_step() (quanda.utils.training.base_pl_module.BasicLightningModule method) (quanda.utils.training.BasicLightningModule method)

W

with_label() (quanda.utils.datasets.dataset_handlers.DatasetHandler method) (quanda.utils.datasets.dataset_handlers.HuggingFaceDatasetHandler method) (quanda.utils.datasets.dataset_handlers.TorchDatasetHandler method)