/RubiksCube-MCTS

Solving the Rubik's cube by training a deep neural network and implementing an MCTS solver

Primary LanguagePython

RubiksCube-MCTS

Solving the Rubik's cube by training a deep neural network and implementing an MCTS solver

Installing env

make develop-pyo3

Training

python train2.py

MCTS Solver

python search.py --load_path "/path/to/checkpoint.pt"