├── embedding ├── action_decoder.py ├── gym_dataset.py ├── main.py ├── main_pixels.py ├── main_vae.py ├── pixel_wrapper.py ├── util.py ├── vae_dyne_action.py ├── vae_dyne_sa.py └── vae_regular.py ├── envs └── reacher_family │ ├── __init__.py │ ├── assets │ ├── materials.xml │ ├── reacher_push.xml │ ├── reacher_turn.xml │ ├── reacher_vertical.xml │ └── visual.xml │ ├── reacher_push.py │ ├── reacher_turn.py │ └── reacher_vertical.py ├── readme.md ├── rl ├── EmbeddedTD3.py ├── PixelTD3.py ├── RandomEmbeddedPolicy.py ├── RandomPolicy.py ├── TD3.py ├── embed_pixel_wrapper.py ├── main_dyne.py ├── main_pixel_td3.py ├── main_td3.py ├── main_vae_td3.py ├── pixel_wrapper.py └── utils.py └── rl_plots.ipynb /embedding/action_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dyne-submission/dynamics-aware-embeddings/HEAD/embedding/action_decoder.py -------------------------------------------------------------------------------- /embedding/gym_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dyne-submission/dynamics-aware-embeddings/HEAD/embedding/gym_dataset.py -------------------------------------------------------------------------------- /embedding/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dyne-submission/dynamics-aware-embeddings/HEAD/embedding/main.py -------------------------------------------------------------------------------- /embedding/main_pixels.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dyne-submission/dynamics-aware-embeddings/HEAD/embedding/main_pixels.py -------------------------------------------------------------------------------- /embedding/main_vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dyne-submission/dynamics-aware-embeddings/HEAD/embedding/main_vae.py -------------------------------------------------------------------------------- /embedding/pixel_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dyne-submission/dynamics-aware-embeddings/HEAD/embedding/pixel_wrapper.py -------------------------------------------------------------------------------- /embedding/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dyne-submission/dynamics-aware-embeddings/HEAD/embedding/util.py -------------------------------------------------------------------------------- /embedding/vae_dyne_action.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dyne-submission/dynamics-aware-embeddings/HEAD/embedding/vae_dyne_action.py -------------------------------------------------------------------------------- /embedding/vae_dyne_sa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dyne-submission/dynamics-aware-embeddings/HEAD/embedding/vae_dyne_sa.py -------------------------------------------------------------------------------- /embedding/vae_regular.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dyne-submission/dynamics-aware-embeddings/HEAD/embedding/vae_regular.py -------------------------------------------------------------------------------- /envs/reacher_family/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dyne-submission/dynamics-aware-embeddings/HEAD/envs/reacher_family/__init__.py -------------------------------------------------------------------------------- /envs/reacher_family/assets/materials.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dyne-submission/dynamics-aware-embeddings/HEAD/envs/reacher_family/assets/materials.xml -------------------------------------------------------------------------------- /envs/reacher_family/assets/reacher_push.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dyne-submission/dynamics-aware-embeddings/HEAD/envs/reacher_family/assets/reacher_push.xml -------------------------------------------------------------------------------- /envs/reacher_family/assets/reacher_turn.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dyne-submission/dynamics-aware-embeddings/HEAD/envs/reacher_family/assets/reacher_turn.xml -------------------------------------------------------------------------------- /envs/reacher_family/assets/reacher_vertical.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dyne-submission/dynamics-aware-embeddings/HEAD/envs/reacher_family/assets/reacher_vertical.xml -------------------------------------------------------------------------------- /envs/reacher_family/assets/visual.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dyne-submission/dynamics-aware-embeddings/HEAD/envs/reacher_family/assets/visual.xml -------------------------------------------------------------------------------- /envs/reacher_family/reacher_push.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dyne-submission/dynamics-aware-embeddings/HEAD/envs/reacher_family/reacher_push.py -------------------------------------------------------------------------------- /envs/reacher_family/reacher_turn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dyne-submission/dynamics-aware-embeddings/HEAD/envs/reacher_family/reacher_turn.py -------------------------------------------------------------------------------- /envs/reacher_family/reacher_vertical.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dyne-submission/dynamics-aware-embeddings/HEAD/envs/reacher_family/reacher_vertical.py -------------------------------------------------------------------------------- /readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dyne-submission/dynamics-aware-embeddings/HEAD/readme.md -------------------------------------------------------------------------------- /rl/EmbeddedTD3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dyne-submission/dynamics-aware-embeddings/HEAD/rl/EmbeddedTD3.py -------------------------------------------------------------------------------- /rl/PixelTD3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dyne-submission/dynamics-aware-embeddings/HEAD/rl/PixelTD3.py -------------------------------------------------------------------------------- /rl/RandomEmbeddedPolicy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dyne-submission/dynamics-aware-embeddings/HEAD/rl/RandomEmbeddedPolicy.py -------------------------------------------------------------------------------- /rl/RandomPolicy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dyne-submission/dynamics-aware-embeddings/HEAD/rl/RandomPolicy.py -------------------------------------------------------------------------------- /rl/TD3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dyne-submission/dynamics-aware-embeddings/HEAD/rl/TD3.py -------------------------------------------------------------------------------- /rl/embed_pixel_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dyne-submission/dynamics-aware-embeddings/HEAD/rl/embed_pixel_wrapper.py -------------------------------------------------------------------------------- /rl/main_dyne.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dyne-submission/dynamics-aware-embeddings/HEAD/rl/main_dyne.py -------------------------------------------------------------------------------- /rl/main_pixel_td3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dyne-submission/dynamics-aware-embeddings/HEAD/rl/main_pixel_td3.py -------------------------------------------------------------------------------- /rl/main_td3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dyne-submission/dynamics-aware-embeddings/HEAD/rl/main_td3.py -------------------------------------------------------------------------------- /rl/main_vae_td3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dyne-submission/dynamics-aware-embeddings/HEAD/rl/main_vae_td3.py -------------------------------------------------------------------------------- /rl/pixel_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dyne-submission/dynamics-aware-embeddings/HEAD/rl/pixel_wrapper.py -------------------------------------------------------------------------------- /rl/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dyne-submission/dynamics-aware-embeddings/HEAD/rl/utils.py -------------------------------------------------------------------------------- /rl_plots.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dyne-submission/dynamics-aware-embeddings/HEAD/rl_plots.ipynb --------------------------------------------------------------------------------