├── .gitignore ├── README.md ├── configs ├── latent_sanity_config.py ├── latent_tello_config.py ├── nolatent_sanity_config.py └── nolatent_tello_config.py ├── requirements.txt ├── scripts ├── eval_model.py ├── eval_policy.py ├── mat_latent_legacy_converter.py ├── plot │ └── plot_model_outputs.py ├── sanity_test.py ├── test.py ├── test_latent_inference.py └── train.py └── src ├── __init__.py └── mbrl ├── __init__.py ├── datasets ├── __init__.py ├── dataset.py ├── mat_dataset.py └── np_dataset.py ├── envs ├── __init__.py ├── env.py ├── env_spec.py ├── latent_env_spec.py ├── ros_copter.py ├── ros_copter_data_capture.py ├── sanity_env.py └── tello_pendulum_controller.py ├── experiments ├── __init__.py ├── file_manager.py └── logger.py ├── models ├── __init__.py ├── gaussian_latent_object.py ├── latent_model.py └── model.py ├── policies ├── __init__.py ├── cem.py ├── latent_mpc_policy.py ├── policy.py ├── random_shooting.py └── ros_policy.py ├── sandbox └── __init__.py ├── trainers ├── __init__.py ├── latent_inference_trainer.py ├── latent_trainer.py └── trainer.py └── utils ├── __init__.py ├── abstract.py ├── data_utils.py ├── file_utils.py ├── mpc_utils.py ├── np_utils.py ├── param_utils.py ├── pyblit.py ├── python_utils.py ├── ros_utils.py ├── tf_utils.py └── torch_utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suneelbelkhale/model-based-meta-rl-for-flight/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suneelbelkhale/model-based-meta-rl-for-flight/HEAD/README.md -------------------------------------------------------------------------------- /configs/latent_sanity_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suneelbelkhale/model-based-meta-rl-for-flight/HEAD/configs/latent_sanity_config.py -------------------------------------------------------------------------------- /configs/latent_tello_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suneelbelkhale/model-based-meta-rl-for-flight/HEAD/configs/latent_tello_config.py -------------------------------------------------------------------------------- /configs/nolatent_sanity_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suneelbelkhale/model-based-meta-rl-for-flight/HEAD/configs/nolatent_sanity_config.py -------------------------------------------------------------------------------- /configs/nolatent_tello_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suneelbelkhale/model-based-meta-rl-for-flight/HEAD/configs/nolatent_tello_config.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suneelbelkhale/model-based-meta-rl-for-flight/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/eval_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suneelbelkhale/model-based-meta-rl-for-flight/HEAD/scripts/eval_model.py -------------------------------------------------------------------------------- /scripts/eval_policy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suneelbelkhale/model-based-meta-rl-for-flight/HEAD/scripts/eval_policy.py -------------------------------------------------------------------------------- /scripts/mat_latent_legacy_converter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suneelbelkhale/model-based-meta-rl-for-flight/HEAD/scripts/mat_latent_legacy_converter.py -------------------------------------------------------------------------------- /scripts/plot/plot_model_outputs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suneelbelkhale/model-based-meta-rl-for-flight/HEAD/scripts/plot/plot_model_outputs.py -------------------------------------------------------------------------------- /scripts/sanity_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suneelbelkhale/model-based-meta-rl-for-flight/HEAD/scripts/sanity_test.py -------------------------------------------------------------------------------- /scripts/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suneelbelkhale/model-based-meta-rl-for-flight/HEAD/scripts/test.py -------------------------------------------------------------------------------- /scripts/test_latent_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suneelbelkhale/model-based-meta-rl-for-flight/HEAD/scripts/test_latent_inference.py -------------------------------------------------------------------------------- /scripts/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suneelbelkhale/model-based-meta-rl-for-flight/HEAD/scripts/train.py -------------------------------------------------------------------------------- /src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mbrl/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mbrl/datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mbrl/datasets/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suneelbelkhale/model-based-meta-rl-for-flight/HEAD/src/mbrl/datasets/dataset.py -------------------------------------------------------------------------------- /src/mbrl/datasets/mat_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suneelbelkhale/model-based-meta-rl-for-flight/HEAD/src/mbrl/datasets/mat_dataset.py -------------------------------------------------------------------------------- /src/mbrl/datasets/np_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suneelbelkhale/model-based-meta-rl-for-flight/HEAD/src/mbrl/datasets/np_dataset.py -------------------------------------------------------------------------------- /src/mbrl/envs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mbrl/envs/env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suneelbelkhale/model-based-meta-rl-for-flight/HEAD/src/mbrl/envs/env.py -------------------------------------------------------------------------------- /src/mbrl/envs/env_spec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suneelbelkhale/model-based-meta-rl-for-flight/HEAD/src/mbrl/envs/env_spec.py -------------------------------------------------------------------------------- /src/mbrl/envs/latent_env_spec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suneelbelkhale/model-based-meta-rl-for-flight/HEAD/src/mbrl/envs/latent_env_spec.py -------------------------------------------------------------------------------- /src/mbrl/envs/ros_copter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suneelbelkhale/model-based-meta-rl-for-flight/HEAD/src/mbrl/envs/ros_copter.py -------------------------------------------------------------------------------- /src/mbrl/envs/ros_copter_data_capture.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suneelbelkhale/model-based-meta-rl-for-flight/HEAD/src/mbrl/envs/ros_copter_data_capture.py -------------------------------------------------------------------------------- /src/mbrl/envs/sanity_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suneelbelkhale/model-based-meta-rl-for-flight/HEAD/src/mbrl/envs/sanity_env.py -------------------------------------------------------------------------------- /src/mbrl/envs/tello_pendulum_controller.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suneelbelkhale/model-based-meta-rl-for-flight/HEAD/src/mbrl/envs/tello_pendulum_controller.py -------------------------------------------------------------------------------- /src/mbrl/experiments/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mbrl/experiments/file_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suneelbelkhale/model-based-meta-rl-for-flight/HEAD/src/mbrl/experiments/file_manager.py -------------------------------------------------------------------------------- /src/mbrl/experiments/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suneelbelkhale/model-based-meta-rl-for-flight/HEAD/src/mbrl/experiments/logger.py -------------------------------------------------------------------------------- /src/mbrl/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mbrl/models/gaussian_latent_object.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suneelbelkhale/model-based-meta-rl-for-flight/HEAD/src/mbrl/models/gaussian_latent_object.py -------------------------------------------------------------------------------- /src/mbrl/models/latent_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suneelbelkhale/model-based-meta-rl-for-flight/HEAD/src/mbrl/models/latent_model.py -------------------------------------------------------------------------------- /src/mbrl/models/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suneelbelkhale/model-based-meta-rl-for-flight/HEAD/src/mbrl/models/model.py -------------------------------------------------------------------------------- /src/mbrl/policies/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mbrl/policies/cem.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suneelbelkhale/model-based-meta-rl-for-flight/HEAD/src/mbrl/policies/cem.py -------------------------------------------------------------------------------- /src/mbrl/policies/latent_mpc_policy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suneelbelkhale/model-based-meta-rl-for-flight/HEAD/src/mbrl/policies/latent_mpc_policy.py -------------------------------------------------------------------------------- /src/mbrl/policies/policy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suneelbelkhale/model-based-meta-rl-for-flight/HEAD/src/mbrl/policies/policy.py -------------------------------------------------------------------------------- /src/mbrl/policies/random_shooting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suneelbelkhale/model-based-meta-rl-for-flight/HEAD/src/mbrl/policies/random_shooting.py -------------------------------------------------------------------------------- /src/mbrl/policies/ros_policy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suneelbelkhale/model-based-meta-rl-for-flight/HEAD/src/mbrl/policies/ros_policy.py -------------------------------------------------------------------------------- /src/mbrl/sandbox/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mbrl/trainers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mbrl/trainers/latent_inference_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suneelbelkhale/model-based-meta-rl-for-flight/HEAD/src/mbrl/trainers/latent_inference_trainer.py -------------------------------------------------------------------------------- /src/mbrl/trainers/latent_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suneelbelkhale/model-based-meta-rl-for-flight/HEAD/src/mbrl/trainers/latent_trainer.py -------------------------------------------------------------------------------- /src/mbrl/trainers/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suneelbelkhale/model-based-meta-rl-for-flight/HEAD/src/mbrl/trainers/trainer.py -------------------------------------------------------------------------------- /src/mbrl/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mbrl/utils/abstract.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suneelbelkhale/model-based-meta-rl-for-flight/HEAD/src/mbrl/utils/abstract.py -------------------------------------------------------------------------------- /src/mbrl/utils/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suneelbelkhale/model-based-meta-rl-for-flight/HEAD/src/mbrl/utils/data_utils.py -------------------------------------------------------------------------------- /src/mbrl/utils/file_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suneelbelkhale/model-based-meta-rl-for-flight/HEAD/src/mbrl/utils/file_utils.py -------------------------------------------------------------------------------- /src/mbrl/utils/mpc_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suneelbelkhale/model-based-meta-rl-for-flight/HEAD/src/mbrl/utils/mpc_utils.py -------------------------------------------------------------------------------- /src/mbrl/utils/np_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suneelbelkhale/model-based-meta-rl-for-flight/HEAD/src/mbrl/utils/np_utils.py -------------------------------------------------------------------------------- /src/mbrl/utils/param_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suneelbelkhale/model-based-meta-rl-for-flight/HEAD/src/mbrl/utils/param_utils.py -------------------------------------------------------------------------------- /src/mbrl/utils/pyblit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suneelbelkhale/model-based-meta-rl-for-flight/HEAD/src/mbrl/utils/pyblit.py -------------------------------------------------------------------------------- /src/mbrl/utils/python_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suneelbelkhale/model-based-meta-rl-for-flight/HEAD/src/mbrl/utils/python_utils.py -------------------------------------------------------------------------------- /src/mbrl/utils/ros_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suneelbelkhale/model-based-meta-rl-for-flight/HEAD/src/mbrl/utils/ros_utils.py -------------------------------------------------------------------------------- /src/mbrl/utils/tf_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suneelbelkhale/model-based-meta-rl-for-flight/HEAD/src/mbrl/utils/tf_utils.py -------------------------------------------------------------------------------- /src/mbrl/utils/torch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suneelbelkhale/model-based-meta-rl-for-flight/HEAD/src/mbrl/utils/torch_utils.py --------------------------------------------------------------------------------