├── .gitignore ├── README.md ├── _assets ├── during_train.gif ├── tensorboard.jpg └── trained.gif ├── envs ├── __init__.py ├── payload_env_g2g.py ├── test.py └── torch_env.py ├── main.py ├── memory.py ├── rollout_generator.py ├── rssm_main_latent_overshooting.py ├── rssm_model.py ├── rssm_policy.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhayraw1/planet-torch/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhayraw1/planet-torch/HEAD/README.md -------------------------------------------------------------------------------- /_assets/during_train.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhayraw1/planet-torch/HEAD/_assets/during_train.gif -------------------------------------------------------------------------------- /_assets/tensorboard.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhayraw1/planet-torch/HEAD/_assets/tensorboard.jpg -------------------------------------------------------------------------------- /_assets/trained.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhayraw1/planet-torch/HEAD/_assets/trained.gif -------------------------------------------------------------------------------- /envs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhayraw1/planet-torch/HEAD/envs/__init__.py -------------------------------------------------------------------------------- /envs/payload_env_g2g.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhayraw1/planet-torch/HEAD/envs/payload_env_g2g.py -------------------------------------------------------------------------------- /envs/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhayraw1/planet-torch/HEAD/envs/test.py -------------------------------------------------------------------------------- /envs/torch_env.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhayraw1/planet-torch/HEAD/main.py -------------------------------------------------------------------------------- /memory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhayraw1/planet-torch/HEAD/memory.py -------------------------------------------------------------------------------- /rollout_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhayraw1/planet-torch/HEAD/rollout_generator.py -------------------------------------------------------------------------------- /rssm_main_latent_overshooting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhayraw1/planet-torch/HEAD/rssm_main_latent_overshooting.py -------------------------------------------------------------------------------- /rssm_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhayraw1/planet-torch/HEAD/rssm_model.py -------------------------------------------------------------------------------- /rssm_policy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhayraw1/planet-torch/HEAD/rssm_policy.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhayraw1/planet-torch/HEAD/utils.py --------------------------------------------------------------------------------