/rubiks-rl

A Rubik's Cube RL environment

Primary LanguagePython

Rubik's Cube RL

Toying around with a Rubik's Cube RL Environment. Uses Stable Baselines for now. Curious to see if multihead attention is a valid features extractor.

Installation

To install:

git clone https://github.com/drubinstein/rubiks-rl.git
pip install -e .

To train

python -m rubiks_rl.train

To mess around with the environment

python -m rubiks_rl.environment