├── .gitignore ├── README.md ├── cfm ├── __init__.py ├── baselines │ ├── planet_models.py │ ├── train_autoencoder.py │ ├── train_joint_dynamics.py │ ├── train_planet.py │ └── train_visual_forward.py ├── dataset.py ├── env │ ├── dm_control_env.py │ ├── spaces │ │ ├── __init__.py │ │ ├── base.py │ │ ├── box.py │ │ ├── composite.py │ │ ├── float_box.py │ │ └── int_box.py │ └── utils.py ├── evaluate_planning.py ├── models.py ├── train_cfm.py ├── utils.py └── visualize │ ├── print_evaluation_stats.py │ └── to_csv.py ├── out └── .gitkeep ├── process_dataset.py ├── requirements.txt ├── run_baselines ├── run_train_autoencoder.py ├── run_train_joint_dynamics.py ├── run_train_planet.py └── run_train_visual_forward.py ├── run_evaluation.py ├── run_train.py ├── sample_trajectories.py └── setup.py /.gitignore: -------------------------------------------------------------------------------- 1 | cfm.egg-info/ 2 | __pycache__/ 3 | data/ 4 | out/ 5 | -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilson1yan/contrastive-forward-model/HEAD/README.md -------------------------------------------------------------------------------- /cfm/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /cfm/baselines/planet_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilson1yan/contrastive-forward-model/HEAD/cfm/baselines/planet_models.py -------------------------------------------------------------------------------- /cfm/baselines/train_autoencoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilson1yan/contrastive-forward-model/HEAD/cfm/baselines/train_autoencoder.py -------------------------------------------------------------------------------- /cfm/baselines/train_joint_dynamics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilson1yan/contrastive-forward-model/HEAD/cfm/baselines/train_joint_dynamics.py -------------------------------------------------------------------------------- /cfm/baselines/train_planet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilson1yan/contrastive-forward-model/HEAD/cfm/baselines/train_planet.py -------------------------------------------------------------------------------- /cfm/baselines/train_visual_forward.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilson1yan/contrastive-forward-model/HEAD/cfm/baselines/train_visual_forward.py -------------------------------------------------------------------------------- /cfm/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilson1yan/contrastive-forward-model/HEAD/cfm/dataset.py -------------------------------------------------------------------------------- /cfm/env/dm_control_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilson1yan/contrastive-forward-model/HEAD/cfm/env/dm_control_env.py -------------------------------------------------------------------------------- /cfm/env/spaces/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /cfm/env/spaces/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilson1yan/contrastive-forward-model/HEAD/cfm/env/spaces/base.py -------------------------------------------------------------------------------- /cfm/env/spaces/box.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilson1yan/contrastive-forward-model/HEAD/cfm/env/spaces/box.py -------------------------------------------------------------------------------- /cfm/env/spaces/composite.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilson1yan/contrastive-forward-model/HEAD/cfm/env/spaces/composite.py -------------------------------------------------------------------------------- /cfm/env/spaces/float_box.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilson1yan/contrastive-forward-model/HEAD/cfm/env/spaces/float_box.py -------------------------------------------------------------------------------- /cfm/env/spaces/int_box.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilson1yan/contrastive-forward-model/HEAD/cfm/env/spaces/int_box.py -------------------------------------------------------------------------------- /cfm/env/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilson1yan/contrastive-forward-model/HEAD/cfm/env/utils.py -------------------------------------------------------------------------------- /cfm/evaluate_planning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilson1yan/contrastive-forward-model/HEAD/cfm/evaluate_planning.py -------------------------------------------------------------------------------- /cfm/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilson1yan/contrastive-forward-model/HEAD/cfm/models.py -------------------------------------------------------------------------------- /cfm/train_cfm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilson1yan/contrastive-forward-model/HEAD/cfm/train_cfm.py -------------------------------------------------------------------------------- /cfm/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilson1yan/contrastive-forward-model/HEAD/cfm/utils.py -------------------------------------------------------------------------------- /cfm/visualize/print_evaluation_stats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilson1yan/contrastive-forward-model/HEAD/cfm/visualize/print_evaluation_stats.py -------------------------------------------------------------------------------- /cfm/visualize/to_csv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilson1yan/contrastive-forward-model/HEAD/cfm/visualize/to_csv.py -------------------------------------------------------------------------------- /out/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /process_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilson1yan/contrastive-forward-model/HEAD/process_dataset.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilson1yan/contrastive-forward-model/HEAD/requirements.txt -------------------------------------------------------------------------------- /run_baselines/run_train_autoencoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilson1yan/contrastive-forward-model/HEAD/run_baselines/run_train_autoencoder.py -------------------------------------------------------------------------------- /run_baselines/run_train_joint_dynamics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilson1yan/contrastive-forward-model/HEAD/run_baselines/run_train_joint_dynamics.py -------------------------------------------------------------------------------- /run_baselines/run_train_planet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilson1yan/contrastive-forward-model/HEAD/run_baselines/run_train_planet.py -------------------------------------------------------------------------------- /run_baselines/run_train_visual_forward.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilson1yan/contrastive-forward-model/HEAD/run_baselines/run_train_visual_forward.py -------------------------------------------------------------------------------- /run_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilson1yan/contrastive-forward-model/HEAD/run_evaluation.py -------------------------------------------------------------------------------- /run_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilson1yan/contrastive-forward-model/HEAD/run_train.py -------------------------------------------------------------------------------- /sample_trajectories.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilson1yan/contrastive-forward-model/HEAD/sample_trajectories.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilson1yan/contrastive-forward-model/HEAD/setup.py --------------------------------------------------------------------------------