.editorconfig
.env.template
.flake8
.gitignore
.pre-commit-config.yaml
LICENSE
README.md
env.yaml
mkdocs.yml
pyproject.toml
setup.cfg
setup.py
.github/workflows/publish.yml
.github/workflows/test_suite.yml
.vscode/launch.json
conf/func_maps.yaml
conf/matching.yaml
conf/matching_n_models.yaml
conf/matching_n_tasks.yaml
conf/merge_n_models.yaml
conf/merge_n_models_scaling.yaml
conf/mlp.yaml
conf/resnet.yaml
conf/split_data_in_tasks.yaml
conf/train_tasks.yaml
conf/vgg.yaml
conf/dataset/cifar10.yaml
conf/dataset/cifar100.yaml
conf/dataset/cifar100_hf.yaml
conf/dataset/cifar100_tasks.yaml
conf/dataset/mnist.yaml
conf/dataset/tiny_imagenet.yaml
conf/hydra/default.yaml
conf/matching/match_n_models.yaml
conf/matching/match_n_tasks.yaml
conf/matching/match_then_sync.yaml
conf/matching/match_then_sync_resnet.yaml
conf/matching/match_two_models.yaml
conf/matching/merge_n_models.yaml
conf/matching/matcher/alternating_diffusion.yaml
conf/matching/matcher/dummy.yaml
conf/matching/matcher/frank_wolfe.yaml
conf/matching/matcher/frank_wolfe_sync.yaml
conf/matching/matcher/git_rebasin.yaml
conf/matching/matcher/quadratic.yaml
conf/matching/matcher/sinkhorn.yaml
conf/matching/matcher/synchronized.yaml
conf/matching/merger/dummy.yaml
conf/matching/merger/frank_wolfe_to_reference.yaml
conf/matching/merger/frank_wolfe_to_universe.yaml
conf/matching/merger/git_rebasin.yaml
conf/matching/merger/git_rebasin_pairwise.yaml
conf/model/mlp.yaml
conf/model/resnet18_pt.yaml
conf/model/resnet20.yaml
conf/model/vgg.yaml
conf/nn/mlp.yaml
conf/nn/n_tasks.yaml
conf/nn/resnet.yaml
conf/nn/vgg.yaml
conf/train/mlp.yaml
conf/train/resnet.yaml
conf/train/resnet18_pt.yaml
conf/train/vgg.yaml
data/.gitignore
docs/index.md
docs/overrides/main.html
notebooks/analyze_match_many.ipynb
notebooks/analyze_merge_many.ipynb
notebooks/demo.ipynb
notebooks/gradient_in_universal_basin.ipynb
notebooks/losses.json
notebooks/merge_models_n_tasks.ipynb
notebooks/quadratic_problem.ipynb
notebooks/repair.ipynb
notebooks/weight_func_maps.ipynb
notebooks/dev/alternating_diffusion.ipynb
notebooks/dev/convert_hf_to_torchvision_dataset.ipynb
notebooks/dev/generalized_dot_product.ipynb
notebooks/dev/gradient_test_perm_indexing.ipynb
notebooks/dev/move_wandb_artifacts.ipynb
notebooks/dev/weight_similarity.ipynb
notebooks/lightning_logs/version_0/events.out.tfevents.1705917548.don-desktop.1374220.0
notebooks/lightning_logs/version_0/events.out.tfevents.1705917551.don-desktop.1374220.1
notebooks/lightning_logs/version_0/events.out.tfevents.1705917555.don-desktop.1374220.2
notebooks/lightning_logs/version_0/events.out.tfevents.1705917558.don-desktop.1374220.3
notebooks/lightning_logs/version_0/events.out.tfevents.1705917561.don-desktop.1374220.4
notebooks/lightning_logs/version_0/events.out.tfevents.1705917565.don-desktop.1374220.5
notebooks/lightning_logs/version_0/events.out.tfevents.1705917568.don-desktop.1374220.6
notebooks/lightning_logs/version_0/events.out.tfevents.1705917571.don-desktop.1374220.7
notebooks/lightning_logs/version_0/events.out.tfevents.1705917575.don-desktop.1374220.8
notebooks/lightning_logs/version_0/events.out.tfevents.1705917578.don-desktop.1374220.9
notebooks/lightning_logs/version_0/events.out.tfevents.1705917584.don-desktop.1374220.10
notebooks/lightning_logs/version_0/events.out.tfevents.1705917588.don-desktop.1374220.11
notebooks/lightning_logs/version_0/events.out.tfevents.1705917592.don-desktop.1374220.12
notebooks/lightning_logs/version_0/hparams.yaml
notebooks/lightning_logs/version_1/events.out.tfevents.1705917641.don-desktop.1382791.0
notebooks/lightning_logs/version_1/events.out.tfevents.1705917645.don-desktop.1382791.1
notebooks/lightning_logs/version_1/events.out.tfevents.1705917648.don-desktop.1382791.2
notebooks/lightning_logs/version_1/events.out.tfevents.1705917652.don-desktop.1382791.3
notebooks/lightning_logs/version_1/events.out.tfevents.1705917655.don-desktop.1382791.4
notebooks/lightning_logs/version_1/events.out.tfevents.1705917658.don-desktop.1382791.5
notebooks/lightning_logs/version_1/events.out.tfevents.1705917662.don-desktop.1382791.6
notebooks/lightning_logs/version_1/events.out.tfevents.1705917665.don-desktop.1382791.7
notebooks/lightning_logs/version_1/events.out.tfevents.1705917668.don-desktop.1382791.8
notebooks/lightning_logs/version_1/events.out.tfevents.1705917672.don-desktop.1382791.9
notebooks/lightning_logs/version_1/events.out.tfevents.1705917675.don-desktop.1382791.10
notebooks/lightning_logs/version_1/events.out.tfevents.1705917679.don-desktop.1382791.11
notebooks/lightning_logs/version_1/events.out.tfevents.1705917682.don-desktop.1382791.12
notebooks/lightning_logs/version_1/events.out.tfevents.1705917685.don-desktop.1382791.13
notebooks/lightning_logs/version_1/events.out.tfevents.1705917689.don-desktop.1382791.14
notebooks/lightning_logs/version_1/events.out.tfevents.1705917693.don-desktop.1382791.15
notebooks/lightning_logs/version_1/events.out.tfevents.1705917696.don-desktop.1382791.16
notebooks/lightning_logs/version_1/events.out.tfevents.1705917699.don-desktop.1382791.17
notebooks/lightning_logs/version_1/events.out.tfevents.1705917703.don-desktop.1382791.18
notebooks/lightning_logs/version_1/events.out.tfevents.1705917706.don-desktop.1382791.19
notebooks/lightning_logs/version_1/events.out.tfevents.1705917710.don-desktop.1382791.20
notebooks/lightning_logs/version_1/events.out.tfevents.1705917714.don-desktop.1382791.21
notebooks/lightning_logs/version_1/events.out.tfevents.1705917717.don-desktop.1382791.22
notebooks/lightning_logs/version_1/events.out.tfevents.1705917721.don-desktop.1382791.23
notebooks/lightning_logs/version_1/events.out.tfevents.1705917724.don-desktop.1382791.24
notebooks/lightning_logs/version_1/events.out.tfevents.1705917728.don-desktop.1382791.25
notebooks/lightning_logs/version_1/events.out.tfevents.1705917731.don-desktop.1382791.26
notebooks/lightning_logs/version_1/events.out.tfevents.1705917735.don-desktop.1382791.27
notebooks/lightning_logs/version_1/events.out.tfevents.1705917738.don-desktop.1382791.28
notebooks/lightning_logs/version_1/events.out.tfevents.1705917742.don-desktop.1382791.29
notebooks/lightning_logs/version_1/hparams.yaml
notebooks/lightning_logs/version_2/events.out.tfevents.1705918025.don-desktop.1382791.30
notebooks/lightning_logs/version_2/events.out.tfevents.1705918028.don-desktop.1382791.31
notebooks/lightning_logs/version_2/events.out.tfevents.1705918032.don-desktop.1382791.32
notebooks/lightning_logs/version_2/events.out.tfevents.1705918035.don-desktop.1382791.33
notebooks/lightning_logs/version_2/events.out.tfevents.1705918039.don-desktop.1382791.34
notebooks/lightning_logs/version_2/events.out.tfevents.1705918042.don-desktop.1382791.35
notebooks/lightning_logs/version_2/events.out.tfevents.1705918046.don-desktop.1382791.36
notebooks/lightning_logs/version_2/events.out.tfevents.1705918049.don-desktop.1382791.37
notebooks/lightning_logs/version_2/events.out.tfevents.1705918053.don-desktop.1382791.38
notebooks/lightning_logs/version_2/events.out.tfevents.1705918056.don-desktop.1382791.39
notebooks/lightning_logs/version_2/events.out.tfevents.1705918060.don-desktop.1382791.40
notebooks/lightning_logs/version_2/events.out.tfevents.1705918063.don-desktop.1382791.41
notebooks/lightning_logs/version_2/events.out.tfevents.1705918066.don-desktop.1382791.42
notebooks/lightning_logs/version_2/events.out.tfevents.1705918070.don-desktop.1382791.43
notebooks/lightning_logs/version_2/events.out.tfevents.1705918073.don-desktop.1382791.44
notebooks/lightning_logs/version_2/events.out.tfevents.1705918077.don-desktop.1382791.45
notebooks/lightning_logs/version_2/events.out.tfevents.1705918080.don-desktop.1382791.46
notebooks/lightning_logs/version_2/events.out.tfevents.1705918083.don-desktop.1382791.47
notebooks/lightning_logs/version_2/events.out.tfevents.1705918087.don-desktop.1382791.48
notebooks/lightning_logs/version_2/events.out.tfevents.1705918090.don-desktop.1382791.49
notebooks/lightning_logs/version_2/events.out.tfevents.1705918093.don-desktop.1382791.50
notebooks/lightning_logs/version_2/events.out.tfevents.1705918097.don-desktop.1382791.51
notebooks/lightning_logs/version_2/events.out.tfevents.1705918101.don-desktop.1382791.52
notebooks/lightning_logs/version_2/events.out.tfevents.1705918104.don-desktop.1382791.53
notebooks/lightning_logs/version_2/events.out.tfevents.1705918108.don-desktop.1382791.54
notebooks/lightning_logs/version_2/events.out.tfevents.1705918111.don-desktop.1382791.55
notebooks/lightning_logs/version_2/events.out.tfevents.1705918115.don-desktop.1382791.56
notebooks/lightning_logs/version_2/events.out.tfevents.1705918118.don-desktop.1382791.57
notebooks/lightning_logs/version_2/events.out.tfevents.1705918121.don-desktop.1382791.58
notebooks/lightning_logs/version_2/events.out.tfevents.1705918125.don-desktop.1382791.59
notebooks/lightning_logs/version_2/hparams.yaml
notebooks/lightning_logs/version_3/events.out.tfevents.1705924624.don-desktop.1382791.60
notebooks/lightning_logs/version_3/events.out.tfevents.1705924627.don-desktop.1382791.61
notebooks/lightning_logs/version_3/hparams.yaml
notebooks/rebuttal/reviewers_baseline.ipynb
output/CIFAR10/VGG16_1.json
output/CIFAR10/VGG16_2.json
output/MNIST/MLP_1.json
output/MNIST/MLP_2.json
shell_scripts/match_and_eval.sh
shell_scripts/move_artifacts.sh
shell_scripts/run_N_seeds.sh
shell_scripts/run_all_approaches.sh
shell_scripts/run_all_mergers.sh
shell_scripts/run_all_pairs.sh
shell_scripts/run_all_seeds.sh
shell_scripts/width_exp.sh
src/ccmm/__init__.py
src/ccmm/_version.py
src/ccmm/data/__init__.py
src/ccmm/data/datamodule.py
src/ccmm/data/datasets.py
src/ccmm/data/my_dataset_dict.py
src/ccmm/data/task_datamodule.py
src/ccmm/matching/__init__.py
src/ccmm/matching/fisher_merging.py
src/ccmm/matching/frank_wolfe_matching.py
src/ccmm/matching/frank_wolfe_sync_matching.py
src/ccmm/matching/matcher.py
src/ccmm/matching/merger.py
src/ccmm/matching/permutation_spec.py
src/ccmm/matching/quadratic_matching.py
src/ccmm/matching/repair.py
src/ccmm/matching/sinkhorn_matching.py
src/ccmm/matching/straight_through_estimator.py
src/ccmm/matching/synchronized_matching.py
src/ccmm/matching/utils.py
src/ccmm/matching/weight_matching.py
src/ccmm/models/__init__.py
src/ccmm/models/cnn.py
src/ccmm/models/mlp.py
src/ccmm/models/repaired_resnet.py
src/ccmm/models/resnet.py
src/ccmm/models/resnet_bn.py
src/ccmm/models/resnet_pt.py
src/ccmm/models/utils.py
src/ccmm/models/vgg.py
src/ccmm/models/vit.py
src/ccmm/modules/__init__.py
src/ccmm/modules/mlp.py
src/ccmm/modules/module.py
src/ccmm/pl_modules/__init__.py
src/ccmm/pl_modules/pl_module.py
src/ccmm/ui/__init__.py
src/ccmm/ui/run.py
src/ccmm/utils/cycle_consistent.py
src/ccmm/utils/graph.py
src/ccmm/utils/io_utils.py
src/ccmm/utils/plot.py
src/ccmm/utils/training.py
src/ccmm/utils/utils.py
src/cycle_consistent_model_merging.egg-info/PKG-INFO
src/cycle_consistent_model_merging.egg-info/SOURCES.txt
src/cycle_consistent_model_merging.egg-info/dependency_links.txt
src/cycle_consistent_model_merging.egg-info/not-zip-safe
src/cycle_consistent_model_merging.egg-info/requires.txt
src/cycle_consistent_model_merging.egg-info/top_level.txt
src/scripts/__init__.py
src/scripts/convert_hf_dataset_to_torchvision.py
src/scripts/create_dummy_models.py
src/scripts/evaluate_matched_models.py
src/scripts/evaluate_merged_model.py
src/scripts/git_rebasin_variance_exp.py
src/scripts/match_n_models.py
src/scripts/match_then_sync.py
src/scripts/match_two_models.py
src/scripts/merge_n_models.py
src/scripts/scaling_merging.py
src/scripts/split_data_in_tasks.py
src/scripts/train.py
src/scripts/train_tasks.py
tests/__init__.py
tests/conftest.py
tests/test_checkpoint.py
tests/test_configuration.py
tests/test_nn_core_integration.py
tests/test_resume.py
tests/test_seeding.py
tests/test_storage.py
tests/test_training.py