vincentoh commited on
Commit
698c5e1
·
verified ·
1 Parent(s): 355ae7a

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +17 -0
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