.gitignore
.pre-commit-config.yaml
.readthedocs.yml
LICENSE.txt
README.md
codecov.yml
mkdocs.yml
pyproject.toml
uv.lock
.github/ISSUE_TEMPLATE/bug-report.md
.github/ISSUE_TEMPLATE/feature-request.md
.github/ISSUE_TEMPLATE/framework-request.md
.github/ISSUE_TEMPLATE/question.md
.github/workflows/lint.yaml
.github/workflows/release.yaml
.github/workflows/test.yaml
disent/__init__.py
disent.egg-info/PKG-INFO
disent.egg-info/SOURCES.txt
disent.egg-info/dependency_links.txt
disent.egg-info/requires.txt
disent.egg-info/scm_file_list.json
disent.egg-info/scm_version.json
disent.egg-info/top_level.txt
disent/dataset/__init__.py
disent/dataset/_base.py
disent/dataset/data/__init__.py
disent/dataset/data/_episodes.py
disent/dataset/data/_episodes__custom.py
disent/dataset/data/_groundtruth.py
disent/dataset/data/_groundtruth__cars3d.py
disent/dataset/data/_groundtruth__dsprites.py
disent/dataset/data/_groundtruth__dsprites_imagenet.py
disent/dataset/data/_groundtruth__mpi3d.py
disent/dataset/data/_groundtruth__norb.py
disent/dataset/data/_groundtruth__shapes3d.py
disent/dataset/data/_groundtruth__sprites.py
disent/dataset/data/_groundtruth__xcolumns.py
disent/dataset/data/_groundtruth__xyobject.py
disent/dataset/data/_groundtruth__xysquares.py
disent/dataset/data/_random__teapots3d.py
disent/dataset/data/_raw.py
disent/dataset/sampling/__init__.py
disent/dataset/sampling/_base.py
disent/dataset/sampling/_groundtruth__dist.py
disent/dataset/sampling/_groundtruth__pair.py
disent/dataset/sampling/_groundtruth__pair_orig.py
disent/dataset/sampling/_groundtruth__single.py
disent/dataset/sampling/_groundtruth__triplet.py
disent/dataset/sampling/_groundtruth__walk.py
disent/dataset/sampling/_random__any.py
disent/dataset/sampling/_random__episodes.py
disent/dataset/sampling/_single.py
disent/dataset/transform/__init__.py
disent/dataset/transform/_augment.py
disent/dataset/transform/_augment_disent.py
disent/dataset/transform/_transforms.py
disent/dataset/transform/functional.py
disent/dataset/util/__init__.py
disent/dataset/util/datafile.py
disent/dataset/util/state_space.py
disent/dataset/util/stats.py
disent/dataset/util/formats/__init__.py
disent/dataset/util/formats/hdf5.py
disent/dataset/util/formats/npz.py
disent/dataset/wrapper/__init__.py
disent/dataset/wrapper/_base.py
disent/dataset/wrapper/_dither.py
disent/dataset/wrapper/_masked.py
disent/frameworks/__init__.py
disent/frameworks/_ae_mixin.py
disent/frameworks/_framework.py
disent/frameworks/ae/__init__.py
disent/frameworks/ae/_supervised__adaneg_tae.py
disent/frameworks/ae/_supervised__tae.py
disent/frameworks/ae/_unsupervised__ae.py
disent/frameworks/ae/_unsupervised__dotae.py
disent/frameworks/ae/_weaklysupervised__adaae.py
disent/frameworks/helper/__init__.py
disent/frameworks/helper/latent_distributions.py
disent/frameworks/helper/reconstructions.py
disent/frameworks/helper/util.py
disent/frameworks/vae/__init__.py
disent/frameworks/vae/_supervised__adaneg_tvae.py
disent/frameworks/vae/_supervised__tvae.py
disent/frameworks/vae/_unsupervised__betatcvae.py
disent/frameworks/vae/_unsupervised__betavae.py
disent/frameworks/vae/_unsupervised__dfcvae.py
disent/frameworks/vae/_unsupervised__dipvae.py
disent/frameworks/vae/_unsupervised__dotvae.py
disent/frameworks/vae/_unsupervised__infovae.py
disent/frameworks/vae/_unsupervised__vae.py
disent/frameworks/vae/_weaklysupervised__adavae.py
disent/metrics/__init__.py
disent/metrics/_dci.py
disent/metrics/_factor_vae.py
disent/metrics/_factored_components.py
disent/metrics/_flatness.py
disent/metrics/_mig.py
disent/metrics/_sap.py
disent/metrics/_unsupervised.py
disent/metrics/utils.py
disent/model/__init__.py
disent/model/_base.py
disent/model/ae/__init__.py
disent/model/ae/_linear.py
disent/model/ae/_norm_conv64.py
disent/model/ae/_vae_conv64.py
disent/model/ae/_vae_fc.py
disent/nn/__init__.py
disent/nn/activations.py
disent/nn/modules.py
disent/nn/weights.py
disent/nn/functional/__init__.py
disent/nn/functional/_conv2d.py
disent/nn/functional/_conv2d_kernels.py
disent/nn/functional/_correlation.py
disent/nn/functional/_dct.py
disent/nn/functional/_mean.py
disent/nn/functional/_norm.py
disent/nn/functional/_other.py
disent/nn/functional/_pca.py
disent/nn/functional/_util_generic.py
disent/nn/loss/__init__.py
disent/nn/loss/kl.py
disent/nn/loss/reduction.py
disent/nn/loss/softsort.py
disent/nn/loss/triplet.py
disent/nn/loss/triplet_mining.py
disent/registry/__init__.py
disent/registry/_registry.py
disent/schedule/__init__.py
disent/schedule/_schedule.py
disent/schedule/lerp.py
disent/util/__init__.py
disent/util/array.py
disent/util/deprecate.py
disent/util/function.py
disent/util/imports.py
disent/util/iters.py
disent/util/jit.py
disent/util/profiling.py
disent/util/seeds.py
disent/util/inout/__init__.py
disent/util/inout/cache.py
disent/util/inout/files.py
disent/util/inout/hashing.py
disent/util/inout/paths.py
disent/util/inout/tar.py
disent/util/lightning/__init__.py
disent/util/lightning/logger_util.py
disent/util/lightning/callbacks/__init__.py
disent/util/lightning/callbacks/_callback_log_metrics.py
disent/util/lightning/callbacks/_callback_print_progress.py
disent/util/lightning/callbacks/_callback_vis_dists.py
disent/util/lightning/callbacks/_callback_vis_latents.py
disent/util/lightning/callbacks/_callbacks_base.py
disent/util/lightning/callbacks/_helper.py
disent/util/math/__init__.py
disent/util/math/dither.py
disent/util/math/integer.py
disent/util/math/random.py
disent/util/strings/__init__.py
disent/util/strings/colors.py
disent/util/strings/fmt.py
disent/util/visualize/__init__.py
disent/util/visualize/plot.py
disent/util/visualize/vis_img.py
disent/util/visualize/vis_latents.py
disent/util/visualize/vis_util.py
docs/index.md
docs/quickstart.md
docs/examples/mnist_example.py
docs/examples/overview_data.py
docs/examples/overview_dataset_loader.py
docs/examples/overview_dataset_pair.py
docs/examples/overview_dataset_pair_augment.py
docs/examples/overview_dataset_single.py
docs/examples/overview_framework_adagvae.py
docs/examples/overview_framework_ae.py
docs/examples/overview_framework_betavae.py
docs/examples/overview_framework_betavae_scheduled.py
docs/examples/overview_framework_train_val.py
docs/examples/overview_metrics.py
docs/examples/readme_example.py
docs/examples/extend_experiment/run.py
docs/examples/extend_experiment/run.sh
docs/examples/extend_experiment/code/__init__.py
docs/examples/extend_experiment/code/groundtruth__xyblocks.py
docs/examples/extend_experiment/code/random_data.py
docs/examples/extend_experiment/code/weaklysupervised__si_adavae.py
docs/examples/extend_experiment/code/weaklysupervised__si_betavae.py
docs/examples/extend_experiment/config/README.md
docs/examples/extend_experiment/config/config_alt.yaml
docs/examples/extend_experiment/config/config_alt_test.yaml
docs/examples/extend_experiment/config/dataset/E--mask-dthr-cars3d.yaml
docs/examples/extend_experiment/config/dataset/E--mask-dthr-dsprites.yaml
docs/examples/extend_experiment/config/dataset/E--mask-dthr-pseudorandom.yaml
docs/examples/extend_experiment/config/dataset/E--mask-dthr-shapes3d.yaml
docs/examples/extend_experiment/config/dataset/E--mask-dthr-smallnorb.yaml
docs/examples/extend_experiment/config/dataset/E--pseudorandom.yaml
docs/examples/extend_experiment/config/dataset/E--xyblocks.yaml
docs/examples/extend_experiment/config/dataset/E--xyblocks_grey.yaml
docs/examples/extend_experiment/config/framework/E--si-adavae.yaml
docs/examples/extend_experiment/config/framework/E--si-betavae.yaml
docs/examples/extend_experiment/config/run_location/cluster_rsync_tmp.yaml
docs/examples/extend_experiment/config/run_location/cluster_shr.yaml
docs/examples/extend_experiment/config/run_location/cluster_tmp.yaml
docs/examples/extend_experiment/config/run_plugins/default.yaml
docs/examples/plotting_examples/__init__.py
docs/examples/plotting_examples/compute_xysquares_overlap.py
docs/examples/plotting_examples/plot_dataset_overlap_diffs.py
docs/examples/plotting_examples/plot_dataset_traversal_dists.py
docs/examples/plotting_examples/plot_dataset_traversals.py
docs/examples/plotting_examples/save_dataset_animation.py
docs/examples/plotting_examples/save_metric_visualisation.py
docs/examples/plotting_examples/plots/.gitignore
docs/examples/plotting_examples/util/__init__.py
docs/examples/plotting_examples/util/gadfly.mplstyle
docs/img/traversals/traversal-transpose__cars3d.jpg
docs/img/traversals/traversal-transpose__dsprites-imagenet-bg-100.jpg
docs/img/traversals/traversal-transpose__dsprites-imagenet-bg-50.jpg
docs/img/traversals/traversal-transpose__dsprites-imagenet-fg-100.jpg
docs/img/traversals/traversal-transpose__dsprites-imagenet-fg-50.jpg
docs/img/traversals/traversal-transpose__dsprites.jpg
docs/img/traversals/traversal-transpose__mpi3d-real.jpg
docs/img/traversals/traversal-transpose__mpi3d-realistic.jpg
docs/img/traversals/traversal-transpose__mpi3d-toy.jpg
docs/img/traversals/traversal-transpose__shapes3d.jpg
docs/img/traversals/traversal-transpose__smallnorb.jpg
docs/img/traversals/traversal-transpose__sprites.jpg
docs/img/traversals/traversal-transpose__xy-blocks.jpg
docs/img/traversals/traversal-transpose__xy-object-shaded.jpg
docs/img/traversals/traversal-transpose__xy-object.jpg
docs/img/traversals/traversal-transpose__xy-single-square__spacing8.jpg
docs/img/traversals/traversal-transpose__xy-squares__spacing8.jpg
experiment/__init__.py
experiment/run.py
experiment/config/__init__.py
experiment/config/config.yaml
experiment/config/config_test.yaml
experiment/config/augment/example.yaml
experiment/config/augment/none.yaml
experiment/config/dataset/cars3d.yaml
experiment/config/dataset/dsprites-imagenet-bg-100.yaml
experiment/config/dataset/dsprites-imagenet-bg-25.yaml
experiment/config/dataset/dsprites-imagenet-bg-50.yaml
experiment/config/dataset/dsprites-imagenet-bg-75.yaml
experiment/config/dataset/dsprites-imagenet-fg-100.yaml
experiment/config/dataset/dsprites-imagenet-fg-25.yaml
experiment/config/dataset/dsprites-imagenet-fg-50.yaml
experiment/config/dataset/dsprites-imagenet-fg-75.yaml
experiment/config/dataset/dsprites-imagenet.yaml
experiment/config/dataset/dsprites.yaml
experiment/config/dataset/mpi3d_real.yaml
experiment/config/dataset/mpi3d_realistic.yaml
experiment/config/dataset/mpi3d_toy.yaml
experiment/config/dataset/shapes3d.yaml
experiment/config/dataset/smallnorb.yaml
experiment/config/dataset/sprites.yaml
experiment/config/dataset/sprites_all.yaml
experiment/config/dataset/xyobject.yaml
experiment/config/dataset/xyobject_grey.yaml
experiment/config/dataset/xyobject_shaded.yaml
experiment/config/dataset/xyobject_shaded_grey.yaml
experiment/config/dataset/xysquares.yaml
experiment/config/dataset/xysquares_grey.yaml
experiment/config/dataset/xysquares_rgb.yaml
experiment/config/dataset/_data_type_/episodes.yaml
experiment/config/dataset/_data_type_/gt.yaml
experiment/config/dataset/_data_type_/random.yaml
experiment/config/framework/adaae.yaml
experiment/config/framework/adaae_os.yaml
experiment/config/framework/adagvae_minimal_os.yaml
experiment/config/framework/adanegtae.yaml
experiment/config/framework/adanegtae_d.yaml
experiment/config/framework/adanegtvae.yaml
experiment/config/framework/adanegtvae_d.yaml
experiment/config/framework/adanegtvae_d_aug.yaml
experiment/config/framework/adavae.yaml
experiment/config/framework/adavae_os.yaml
experiment/config/framework/ae.yaml
experiment/config/framework/betatcvae.yaml
experiment/config/framework/betavae.yaml
experiment/config/framework/dfcvae.yaml
experiment/config/framework/dipvae.yaml
experiment/config/framework/infovae.yaml
experiment/config/framework/tae.yaml
experiment/config/framework/tvae.yaml
experiment/config/framework/vae.yaml
experiment/config/framework/_input_mode_/pair.yaml
experiment/config/framework/_input_mode_/single.yaml
experiment/config/framework/_input_mode_/triplet.yaml
experiment/config/framework/_input_mode_/weak_pair.yaml
experiment/config/metrics/all.yaml
experiment/config/metrics/fast.yaml
experiment/config/metrics/none.yaml
experiment/config/metrics/test.yaml
experiment/config/model/linear.yaml
experiment/config/model/norm_conv64.yaml
experiment/config/model/vae_conv64.yaml
experiment/config/model/vae_fc.yaml
experiment/config/optimizer/adabelief.yaml
experiment/config/optimizer/adam.yaml
experiment/config/optimizer/amsgrad.yaml
experiment/config/optimizer/radam.yaml
experiment/config/optimizer/rmsprop.yaml
experiment/config/optimizer/sgd.yaml
experiment/config/run_action/prepare_data.yaml
experiment/config/run_action/skip.yaml
experiment/config/run_action/train.yaml
experiment/config/run_callbacks/all.yaml
experiment/config/run_callbacks/none.yaml
experiment/config/run_callbacks/test.yaml
experiment/config/run_callbacks/vis.yaml
experiment/config/run_callbacks/vis_debug.yaml
experiment/config/run_callbacks/vis_fast.yaml
experiment/config/run_callbacks/vis_quick.yaml
experiment/config/run_callbacks/vis_skip_first.yaml
experiment/config/run_callbacks/vis_slow.yaml
experiment/config/run_launcher/local.yaml
experiment/config/run_launcher/slurm.yaml
experiment/config/run_length/debug.yaml
experiment/config/run_length/epic.yaml
experiment/config/run_length/long.yaml
experiment/config/run_length/longmed.yaml
experiment/config/run_length/medium.yaml
experiment/config/run_length/short.yaml
experiment/config/run_length/test.yaml
experiment/config/run_length/tiny.yaml
experiment/config/run_length/xtiny.yaml
experiment/config/run_location/local.yaml
experiment/config/run_location/local_cpu.yaml
experiment/config/run_location/local_gpu.yaml
experiment/config/run_logging/none.yaml
experiment/config/run_logging/wandb.yaml
experiment/config/run_logging/wandb_fast.yaml
experiment/config/run_logging/wandb_fast_offline.yaml
experiment/config/run_logging/wandb_slow.yaml
experiment/config/run_plugins/default.yaml
experiment/config/sampling/default.yaml
experiment/config/sampling/default__bb.yaml
experiment/config/sampling/default__ran_l1.yaml
experiment/config/sampling/default__ran_l2.yaml
experiment/config/sampling/gt_dist__combined.yaml
experiment/config/sampling/gt_dist__combined_scaled.yaml
experiment/config/sampling/gt_dist__factors.yaml
experiment/config/sampling/gt_dist__manhat.yaml
experiment/config/sampling/gt_dist__manhat_scaled.yaml
experiment/config/sampling/gt_dist__random.yaml
experiment/config/sampling/none.yaml
experiment/config/sampling/random.yaml
experiment/config/sampling/_sampler_/episodes__pair.yaml
experiment/config/sampling/_sampler_/episodes__single.yaml
experiment/config/sampling/_sampler_/episodes__triplet.yaml
experiment/config/sampling/_sampler_/episodes__weak_pair.yaml
experiment/config/sampling/_sampler_/gt__pair.yaml
experiment/config/sampling/_sampler_/gt__single.yaml
experiment/config/sampling/_sampler_/gt__triplet.yaml
experiment/config/sampling/_sampler_/gt__weak_pair.yaml
experiment/config/sampling/_sampler_/gt_dist__pair.yaml
experiment/config/sampling/_sampler_/gt_dist__single.yaml
experiment/config/sampling/_sampler_/gt_dist__triplet.yaml
experiment/config/sampling/_sampler_/gt_dist__weak_pair.yaml
experiment/config/sampling/_sampler_/random__pair.yaml
experiment/config/sampling/_sampler_/random__single.yaml
experiment/config/sampling/_sampler_/random__triplet.yaml
experiment/config/sampling/_sampler_/random__weak_pair.yaml
experiment/config/schedule/adanegtvae_up_all.yaml
experiment/config/schedule/adanegtvae_up_all_full.yaml
experiment/config/schedule/adanegtvae_up_all_weak.yaml
experiment/config/schedule/adanegtvae_up_ratio.yaml
experiment/config/schedule/adanegtvae_up_ratio_full.yaml
experiment/config/schedule/adanegtvae_up_ratio_weak.yaml
experiment/config/schedule/adanegtvae_up_thresh.yaml
experiment/config/schedule/beta_cyclic.yaml
experiment/config/schedule/beta_cyclic_fast.yaml
experiment/config/schedule/beta_cyclic_slow.yaml
experiment/config/schedule/beta_decrease.yaml
experiment/config/schedule/beta_delay.yaml
experiment/config/schedule/beta_delay_long.yaml
experiment/config/schedule/beta_increase.yaml
experiment/config/schedule/none.yaml
experiment/util/__init__.py
experiment/util/hydra_data.py
experiment/util/hydra_main.py
experiment/util/hydra_utils.py
experiment/util/path_utils.py
experiment/util/run_utils.py
experiment/util/_hydra_searchpath_plugin_/hydra_plugins/searchpath_plugin.py
tests/__init__.py
tests/test_000_import.py
tests/test_data_similarity.py
tests/test_dataset_formats.py
tests/test_docs_examples.py
tests/test_experiment.py
tests/test_frameworks.py
tests/test_math.py
tests/test_math_generic.py
tests/test_metrics.py
tests/test_models.py
tests/test_registry.py
tests/test_samplers.py
tests/test_state_space.py
tests/test_to_img.py
tests/test_transform.py
tests/util.py