Latent Go-Explore
Official implementation of Latent Go-Explore (LGE) algorithm.
Paper: Cell-Free Latent Go-Explore
Installation
git clone https://github.com/qgallouedec/lge
cd lge
pip install -e .
Usage
from stable_baselines3 import SAC
from lge import LatentGoExplore
lge = LatentGoExplore(SAC, "MountainCarContinuous-v0")
lge.explore(total_timesteps=10_000)
Or via command line
python experiments/explore_lge.py --env MountainCarContinuous-v0 --algo sac
Supported envrionments specifications:
Space | Observation space |
---|---|
Discrete |
✔️ |
Box |
✔️ |
Image | ✔️ |
MultiDiscrete |
❌ |
MultiBinary |
✔️ |
Space | Action space |
---|---|
Box |
✔️ |
Discrete |
✔️ |
Image is actually a multi-dimensonal uint8 Box
.