alexever commited on
Commit
c29ede7
·
verified ·
1 Parent(s): adeee2a

Add model card

Browse files
Files changed (1) hide show
  1. README.md +46 -0
README.md ADDED
@@ -0,0 +1,46 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ library_name: pytorch
4
+ tags:
5
+ - rubiks-cube
6
+ - reinforcement-learning
7
+ - deepcubea
8
+ - pytorch
9
+ pipeline_tag: other
10
+ ---
11
+
12
+ # DeepCube — Cube3 (3×3×3) cost-to-go network
13
+
14
+ PyTorch weights for a DeepCubeA-style cost-to-go network that solves the 3×3×3 Rubik's cube via weighted A\*.
15
+
16
+ - **Input**: one-hot encoded cube state (324 dims = 54 stickers × 6 colors)
17
+ - **Output**: scalar cost-to-go estimate (predicted moves to solved state)
18
+ - **Architecture**: MLP — see `deepcube/model.py` in the source repo
19
+ - **Training**: Approximate Value Iteration on random scrambles (see `train.ipynb`)
20
+ - **Source code**: https://github.com/ac1982/deepcube
21
+
22
+ ## Files
23
+
24
+ - `deepcube_cube3.pt` — final weights (keys: `net`, `cfg`, `loss_hist`, `elapsed`)
25
+
26
+ ## Usage
27
+
28
+ ```python
29
+ from huggingface_hub import hf_hub_download
30
+ from deepcube.model import load_checkpoint
31
+
32
+ ckpt_path = hf_hub_download("alexever/deepcube-cube3", "deepcube_cube3.pt")
33
+ model, cfg = load_checkpoint(ckpt_path)
34
+ ```
35
+
36
+ Or via the bundled server, which auto-loads from `checkpoints/deepcube_cube3.pt`:
37
+
38
+ ```bash
39
+ huggingface-cli download alexever/deepcube-cube3 deepcube_cube3.pt \
40
+ --local-dir checkpoints
41
+ python -m deepcube.server
42
+ ```
43
+
44
+ ## License
45
+
46
+ MIT — see source repository.