Skip to content
/ PlaNet Public

Deep Planning Network: Control from pixels by latent planning with learned dynamics

License

Notifications You must be signed in to change notification settings

Kaixhin/PlaNet

Repository files navigation

PlaNet

MIT License

PlaNet: A Deep Planning Network for Reinforcement Learning [1]. Supports symbolic/visual observation spaces. Supports some Gym environments (including classic control/non-MuJoCo environments, so DeepMind Control Suite/MuJoCo are optional dependencies). Hyperparameters have been taken from the original work and are tuned for DeepMind Control Suite, so would need tuning for any other domains (such as the Gym environments).

Run with python.main.py. For best performance with DeepMind Control Suite, try setting environment variable MUJOCO_GL=egl (see instructions and details here).

Results and pretrained models can be found in the releases.

Requirements

To install all dependencies with Anaconda run conda env create -f environment.yml and use source activate planet to activate the environment.

Links

Acknowledgements

References

[1] Learning Latent Dynamics for Planning from Pixels

About

Deep Planning Network: Control from pixels by latent planning with learned dynamics

Topics

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Contributors 4

  •  
  •  
  •  
  •  

Languages