fuyw / lge

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.