├── .gitignore ├── CITATION.cff ├── README.md ├── assets ├── explicit_mse_10.png ├── explicit_mse_30.png ├── implicit_ebm_10.png └── implicit_ebm_30.png ├── ibc ├── __init__.py ├── dataset.py ├── experiment.py ├── models.py ├── modules.py ├── optimizers.py ├── trainer.py └── utils.py ├── mypy.ini ├── plot.py ├── requirements.txt ├── run_explicit.sh ├── run_implicit.sh └── train.py /.gitignore: -------------------------------------------------------------------------------- 1 | __pycache__/ 2 | experiments/ 3 | -------------------------------------------------------------------------------- /CITATION.cff: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kevinzakka/ibc/HEAD/CITATION.cff -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kevinzakka/ibc/HEAD/README.md -------------------------------------------------------------------------------- /assets/explicit_mse_10.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kevinzakka/ibc/HEAD/assets/explicit_mse_10.png -------------------------------------------------------------------------------- /assets/explicit_mse_30.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kevinzakka/ibc/HEAD/assets/explicit_mse_30.png -------------------------------------------------------------------------------- /assets/implicit_ebm_10.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kevinzakka/ibc/HEAD/assets/implicit_ebm_10.png -------------------------------------------------------------------------------- /assets/implicit_ebm_30.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kevinzakka/ibc/HEAD/assets/implicit_ebm_30.png -------------------------------------------------------------------------------- /ibc/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ibc/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kevinzakka/ibc/HEAD/ibc/dataset.py -------------------------------------------------------------------------------- /ibc/experiment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kevinzakka/ibc/HEAD/ibc/experiment.py -------------------------------------------------------------------------------- /ibc/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kevinzakka/ibc/HEAD/ibc/models.py -------------------------------------------------------------------------------- /ibc/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kevinzakka/ibc/HEAD/ibc/modules.py -------------------------------------------------------------------------------- /ibc/optimizers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kevinzakka/ibc/HEAD/ibc/optimizers.py -------------------------------------------------------------------------------- /ibc/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kevinzakka/ibc/HEAD/ibc/trainer.py -------------------------------------------------------------------------------- /ibc/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kevinzakka/ibc/HEAD/ibc/utils.py -------------------------------------------------------------------------------- /mypy.ini: -------------------------------------------------------------------------------- 1 | [mypy] 2 | ignore_missing_imports = True 3 | -------------------------------------------------------------------------------- /plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kevinzakka/ibc/HEAD/plot.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kevinzakka/ibc/HEAD/requirements.txt -------------------------------------------------------------------------------- /run_explicit.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kevinzakka/ibc/HEAD/run_explicit.sh -------------------------------------------------------------------------------- /run_implicit.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kevinzakka/ibc/HEAD/run_implicit.sh -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kevinzakka/ibc/HEAD/train.py --------------------------------------------------------------------------------