Upload README.md with huggingface_hub
Browse files
README.md
CHANGED
|
@@ -40,6 +40,7 @@ CTM's key innovation: **accuracy improves with more internal iterations**. The m
|
|
| 40 |
| Tracking-Position | `ctm-tracking-position.pt` | 6.7M | Exact position | 93.8% | 256-class (16x16 grid) |
|
| 41 |
| Transfer | `ctm-transfer-parity-brackets.pt` | 2.5M | Transfer learning | 94.5% | Parity core to brackets |
|
| 42 |
| Jigsaw MNIST | `ctm-jigsaw-mnist.pt` | 19M | Jigsaw puzzle solving | 92.3% | Reassemble 2x2 shuffled MNIST |
|
|
|
|
| 43 |
|
| 44 |
## Model Configurations
|
| 45 |
|
|
@@ -129,6 +130,22 @@ config = {
|
|
| 129 |
}
|
| 130 |
```
|
| 131 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 132 |
## Usage
|
| 133 |
|
| 134 |
```python
|
|
|
|
| 40 |
| Tracking-Position | `ctm-tracking-position.pt` | 6.7M | Exact position | 93.8% | 256-class (16x16 grid) |
|
| 41 |
| Transfer | `ctm-transfer-parity-brackets.pt` | 2.5M | Transfer learning | 94.5% | Parity core to brackets |
|
| 42 |
| Jigsaw MNIST | `ctm-jigsaw-mnist.pt` | 19M | Jigsaw puzzle solving | 92.3% | Reassemble 2x2 shuffled MNIST |
|
| 43 |
+
| Rotation MNIST | `ctm-rotation-mnist.pt` | 4.2M | Rotation prediction | 89.1% | Predict rotation angle (4 classes) |
|
| 44 |
|
| 45 |
## Model Configurations
|
| 46 |
|
|
|
|
| 130 |
}
|
| 131 |
```
|
| 132 |
|
| 133 |
+
### Rotation MNIST CTM
|
| 134 |
+
```python
|
| 135 |
+
config = {
|
| 136 |
+
"iterations": 20,
|
| 137 |
+
"memory_length": 15,
|
| 138 |
+
"d_model": 256,
|
| 139 |
+
"d_input": 64,
|
| 140 |
+
"heads": 4,
|
| 141 |
+
"n_synch_out": 32,
|
| 142 |
+
"n_synch_action": 32,
|
| 143 |
+
"synapse_depth": 1,
|
| 144 |
+
"out_dims": 4, # 0°, 90°, 180°, 270°
|
| 145 |
+
"backbone_type": "rotation",
|
| 146 |
+
}
|
| 147 |
+
```
|
| 148 |
+
|
| 149 |
## Usage
|
| 150 |
|
| 151 |
```python
|