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:

Image is actually a multi-dimensonal uint8 Box.