├── .gitignore ├── LICENSE.md ├── README.md ├── notebooks ├── generalisation.ipynb ├── postproc.ipynb └── transformers.ipynb ├── scripts ├── analysis │ ├── conditions.yaml │ ├── disent.py │ ├── exportdb.py │ └── utils.py ├── bin │ ├── run-composition.sh │ ├── run-decoders.sh │ └── run-disent.sh ├── configs │ ├── cnnvae.py │ ├── composers.py │ ├── datasplits.py │ ├── decoders.py │ └── training.py ├── experiments │ ├── composition.py │ ├── decoders.py │ └── disent.py └── ingredients │ ├── autoencoders.py │ ├── composers.py │ ├── dataset.py │ ├── decoders.py │ └── training.py ├── src ├── analysis │ ├── hinton.py │ ├── metrics.py │ ├── testing.py │ └── traversal.py ├── dataset │ ├── mpi.py │ ├── shapes3d.py │ ├── sprites.py │ └── transforms.py ├── models │ ├── feedforward.py │ ├── initialization.py │ ├── lgm.py │ └── stochastic.py └── training │ ├── handlers.py │ ├── loss.py │ └── optimizer.py └── torchlab-env.yml /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmrl/disent-and-gen/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmrl/disent-and-gen/HEAD/LICENSE.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmrl/disent-and-gen/HEAD/README.md -------------------------------------------------------------------------------- /notebooks/generalisation.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmrl/disent-and-gen/HEAD/notebooks/generalisation.ipynb -------------------------------------------------------------------------------- /notebooks/postproc.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmrl/disent-and-gen/HEAD/notebooks/postproc.ipynb -------------------------------------------------------------------------------- /notebooks/transformers.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmrl/disent-and-gen/HEAD/notebooks/transformers.ipynb -------------------------------------------------------------------------------- /scripts/analysis/conditions.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmrl/disent-and-gen/HEAD/scripts/analysis/conditions.yaml -------------------------------------------------------------------------------- /scripts/analysis/disent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmrl/disent-and-gen/HEAD/scripts/analysis/disent.py -------------------------------------------------------------------------------- /scripts/analysis/exportdb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmrl/disent-and-gen/HEAD/scripts/analysis/exportdb.py -------------------------------------------------------------------------------- /scripts/analysis/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmrl/disent-and-gen/HEAD/scripts/analysis/utils.py -------------------------------------------------------------------------------- /scripts/bin/run-composition.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmrl/disent-and-gen/HEAD/scripts/bin/run-composition.sh -------------------------------------------------------------------------------- /scripts/bin/run-decoders.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmrl/disent-and-gen/HEAD/scripts/bin/run-decoders.sh -------------------------------------------------------------------------------- /scripts/bin/run-disent.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmrl/disent-and-gen/HEAD/scripts/bin/run-disent.sh -------------------------------------------------------------------------------- /scripts/configs/cnnvae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmrl/disent-and-gen/HEAD/scripts/configs/cnnvae.py -------------------------------------------------------------------------------- /scripts/configs/composers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmrl/disent-and-gen/HEAD/scripts/configs/composers.py -------------------------------------------------------------------------------- /scripts/configs/datasplits.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmrl/disent-and-gen/HEAD/scripts/configs/datasplits.py -------------------------------------------------------------------------------- /scripts/configs/decoders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmrl/disent-and-gen/HEAD/scripts/configs/decoders.py -------------------------------------------------------------------------------- /scripts/configs/training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmrl/disent-and-gen/HEAD/scripts/configs/training.py -------------------------------------------------------------------------------- /scripts/experiments/composition.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmrl/disent-and-gen/HEAD/scripts/experiments/composition.py -------------------------------------------------------------------------------- /scripts/experiments/decoders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmrl/disent-and-gen/HEAD/scripts/experiments/decoders.py -------------------------------------------------------------------------------- /scripts/experiments/disent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmrl/disent-and-gen/HEAD/scripts/experiments/disent.py -------------------------------------------------------------------------------- /scripts/ingredients/autoencoders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmrl/disent-and-gen/HEAD/scripts/ingredients/autoencoders.py -------------------------------------------------------------------------------- /scripts/ingredients/composers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmrl/disent-and-gen/HEAD/scripts/ingredients/composers.py -------------------------------------------------------------------------------- /scripts/ingredients/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmrl/disent-and-gen/HEAD/scripts/ingredients/dataset.py -------------------------------------------------------------------------------- /scripts/ingredients/decoders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmrl/disent-and-gen/HEAD/scripts/ingredients/decoders.py -------------------------------------------------------------------------------- /scripts/ingredients/training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmrl/disent-and-gen/HEAD/scripts/ingredients/training.py -------------------------------------------------------------------------------- /src/analysis/hinton.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmrl/disent-and-gen/HEAD/src/analysis/hinton.py -------------------------------------------------------------------------------- /src/analysis/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmrl/disent-and-gen/HEAD/src/analysis/metrics.py -------------------------------------------------------------------------------- /src/analysis/testing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmrl/disent-and-gen/HEAD/src/analysis/testing.py -------------------------------------------------------------------------------- /src/analysis/traversal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmrl/disent-and-gen/HEAD/src/analysis/traversal.py -------------------------------------------------------------------------------- /src/dataset/mpi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmrl/disent-and-gen/HEAD/src/dataset/mpi.py -------------------------------------------------------------------------------- /src/dataset/shapes3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmrl/disent-and-gen/HEAD/src/dataset/shapes3d.py -------------------------------------------------------------------------------- /src/dataset/sprites.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmrl/disent-and-gen/HEAD/src/dataset/sprites.py -------------------------------------------------------------------------------- /src/dataset/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmrl/disent-and-gen/HEAD/src/dataset/transforms.py -------------------------------------------------------------------------------- /src/models/feedforward.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmrl/disent-and-gen/HEAD/src/models/feedforward.py -------------------------------------------------------------------------------- /src/models/initialization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmrl/disent-and-gen/HEAD/src/models/initialization.py -------------------------------------------------------------------------------- /src/models/lgm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmrl/disent-and-gen/HEAD/src/models/lgm.py -------------------------------------------------------------------------------- /src/models/stochastic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmrl/disent-and-gen/HEAD/src/models/stochastic.py -------------------------------------------------------------------------------- /src/training/handlers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmrl/disent-and-gen/HEAD/src/training/handlers.py -------------------------------------------------------------------------------- /src/training/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmrl/disent-and-gen/HEAD/src/training/loss.py -------------------------------------------------------------------------------- /src/training/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmrl/disent-and-gen/HEAD/src/training/optimizer.py -------------------------------------------------------------------------------- /torchlab-env.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmrl/disent-and-gen/HEAD/torchlab-env.yml --------------------------------------------------------------------------------