LICENSE
README.md
pyproject.toml
setup.py
dattri/__init__.py
dattri/task.py
dattri.egg-info/PKG-INFO
dattri.egg-info/SOURCES.txt
dattri.egg-info/dependency_links.txt
dattri.egg-info/entry_points.txt
dattri.egg-info/requires.txt
dattri.egg-info/top_level.txt
dattri/algorithm/__init__.py
dattri/algorithm/base.py
dattri/algorithm/data_shapley.py
dattri/algorithm/dvemb.py
dattri/algorithm/factgrass.py
dattri/algorithm/influence_function.py
dattri/algorithm/logra.py
dattri/algorithm/rps.py
dattri/algorithm/tracin.py
dattri/algorithm/trak.py
dattri/algorithm/utils.py
dattri/algorithm/block_projected_if/__init__.py
dattri/algorithm/block_projected_if/block_projected_if.py
dattri/algorithm/block_projected_if/core/__init__.py
dattri/algorithm/block_projected_if/core/compressor.py
dattri/algorithm/block_projected_if/core/hook.py
dattri/algorithm/block_projected_if/core/metadata.py
dattri/algorithm/block_projected_if/core/utils.py
dattri/algorithm/block_projected_if/offload/__init__.py
dattri/algorithm/block_projected_if/offload/cpu.py
dattri/algorithm/block_projected_if/offload/disk.py
dattri/algorithm/block_projected_if/offload/memory.py
dattri/algorithm/block_projected_if/offload/offload.py
dattri/algorithm/block_projected_if/offload/disk_io/__init__.py
dattri/algorithm/block_projected_if/offload/disk_io/manager.py
dattri/algorithm/block_projected_if/offload/disk_io/memory_map.py
dattri/algorithm/block_projected_if/offload/disk_io/prefetch_dataset.py
dattri/benchmark/__init__.py
dattri/benchmark/load.py
dattri/benchmark/utils.py
dattri/benchmark/datasets/__init__.py
dattri/benchmark/datasets/cifar/__init__.py
dattri/benchmark/datasets/cifar/cifar_resnet9.py
dattri/benchmark/datasets/cifar/data.py
dattri/benchmark/datasets/imagenet/__init__.py
dattri/benchmark/datasets/imagenet/data.py
dattri/benchmark/datasets/imagenet/imagenet_resnet18.py
dattri/benchmark/datasets/maestro/__init__.py
dattri/benchmark/datasets/maestro/data.py
dattri/benchmark/datasets/maestro/maestro_musictransformer.py
dattri/benchmark/datasets/mnist/__init__.py
dattri/benchmark/datasets/mnist/data.py
dattri/benchmark/datasets/mnist/mnist_lr.py
dattri/benchmark/datasets/mnist/mnist_mlp.py
dattri/benchmark/datasets/shakespeare_char/__init__.py
dattri/benchmark/datasets/shakespeare_char/data.py
dattri/benchmark/datasets/shakespeare_char/prepare.py
dattri/benchmark/datasets/tinystories/prepare.py
dattri/benchmark/datasets/wikitext2/__init__.py
dattri/benchmark/datasets/wikitext2/data.py
dattri/benchmark/datasets/wikitext2/wikitext2_gpt2.py
dattri/benchmark/models/__init__.py
dattri/benchmark/models/MusicTransformer/__init__.py
dattri/benchmark/models/MusicTransformer/generate.py
dattri/benchmark/models/MusicTransformer/loss.py
dattri/benchmark/models/MusicTransformer/music_transformer.py
dattri/benchmark/models/MusicTransformer/positional_encoding.py
dattri/benchmark/models/MusicTransformer/rpr.py
dattri/benchmark/models/MusicTransformer/dataset/__init__.py
dattri/benchmark/models/MusicTransformer/dataset/e_piano.py
dattri/benchmark/models/MusicTransformer/dataset/preprocess_midi.py
dattri/benchmark/models/MusicTransformer/midi_processor/__init__.py
dattri/benchmark/models/MusicTransformer/midi_processor/processor.py
dattri/benchmark/models/MusicTransformer/utilities/__init__.py
dattri/benchmark/models/MusicTransformer/utilities/argument_funcs.py
dattri/benchmark/models/MusicTransformer/utilities/constants.py
dattri/benchmark/models/MusicTransformer/utilities/device.py
dattri/benchmark/models/MusicTransformer/utilities/lr_scheduling.py
dattri/benchmark/models/MusicTransformer/utilities/run_model.py
dattri/benchmark/models/gpt2/__init__.py
dattri/benchmark/models/gpt2/gpt2.py
dattri/benchmark/models/logistic_regression/__init__.py
dattri/benchmark/models/logistic_regression/logistic_regression.py
dattri/benchmark/models/mlp/__init__.py
dattri/benchmark/models/mlp/mlp.py
dattri/benchmark/models/nanoGPT/configurator.py
dattri/benchmark/models/nanoGPT/model.py
dattri/benchmark/models/nanoGPT/train.py
dattri/benchmark/models/nanoGPT/config/train_shakespeare_char.py
dattri/benchmark/models/nanoGPT/config/train_tinystories.py
dattri/benchmark/models/resnet9/__init__.py
dattri/benchmark/models/resnet9/resnet9.py
dattri/func/__init__.py
dattri/func/fisher.py
dattri/func/hessian.py
dattri/func/projection.py
dattri/func/utils.py
dattri/metric/__init__.py
dattri/metric/britteness.py
dattri/metric/ground_truth.py
dattri/metric/metrics.py
dattri/model_util/__init__.py
dattri/model_util/dropout.py
dattri/model_util/ensemble.py
dattri/model_util/hook.py
dattri/model_util/retrain.py
dattri/script/__init__.py
dattri/script/dattri_retrain.py
dattri/script/dattri_retrain_nanogpt.py
docs/source/conf.py
examples/brittleness/mnist_lr_brittleness.py
examples/customized_retraining/mnist.py
examples/data_cleaning/influence_function_data_cleaning.py
examples/dvemb/_groundtruth_loo_mlp.py
examples/dvemb/dvemb_mlp.py
examples/noisy_label_detection/influence_function_noisy_label.py
examples/noisy_label_detection/tracin_noisy_label.py
examples/noisy_label_detection/trak_noisy_label.py
examples/pretrained_benchmark/influence_function_lds.py
examples/pretrained_benchmark/logra_mnist_mlp_lds.py
examples/pretrained_benchmark/logra_wikitext2_gpt2_lds.py
examples/pretrained_benchmark/trak_dropout_lds.py
examples/pretrained_benchmark/trak_loo.py
examples/relatIF/influence_function_comparison.py
experiments/benchmark_result.py
experiments/benchmark_result_mt.py
experiments/benchmark_result_nanogpt.py
experiments/gpt2_wikitext/groundtruth.py
experiments/gpt2_wikitext/score_TRAK.py
experiments/gpt2_wikitext/score_logra.py
experiments/gpt2_wikitext/spearman.py
experiments/gpt2_wikitext/train.py
test/dattri/test_task.py
test/dattri/algorithm/test_data_shapley.py
test/dattri/algorithm/test_dvemb.py
test/dattri/algorithm/test_factgrass.py
test/dattri/algorithm/test_influence_function.py
test/dattri/algorithm/test_logra.py
test/dattri/algorithm/test_rps.py
test/dattri/algorithm/test_tracin.py
test/dattri/algorithm/test_trak.py
test/dattri/benchmark/test_cifar2.py
test/dattri/benchmark/test_flip_label.py
test/dattri/benchmark/test_imagenet.py
test/dattri/benchmark/test_maestro.py
test/dattri/benchmark/test_mnist.py
test/dattri/benchmark/test_utils.py
test/dattri/func/test_fisher.py
test/dattri/func/test_hessian.py
test/dattri/func/test_proj.py
test/dattri/metrics/test_lds.py
test/dattri/metrics/test_loo.py
test/dattri/metrics/test_mislabel_detection_auc.py
test/dattri/model_utils/test_dropout.py
test/dattri/model_utils/test_hook.py
test/dattri/model_utils/test_retrain.py