cube-solver-pytorch.ipynb is a jupyter notebook explaining how to build the model. cube-solver-manual.ipynb is a take at implementing the processus from scratch without using PyTorch.nn.
rlepretre / cube-solver Goto Github PK
View Code? Open in Web Editor NEWMy attempt at training a reinforcement learning agent at solving 2x2 rubik's cube using DQN (implemented with PyTorch) and a custom made Gymnasium environment.