PhilipQuirke
commited on
Update model card with revised readme
Browse files
README.md
CHANGED
@@ -1,15 +1,24 @@
|
|
1 |
# QuantaMaths: `add_d15_l2_h3_t80K_s572091`
|
2 |
|
3 |
-
|
4 |
|
5 |
-
-
|
6 |
-
- **
|
7 |
-
- **
|
8 |
-
- **
|
9 |
-
- **
|
10 |
-
- **
|
|
|
|
|
|
|
|
|
|
|
|
|
11 |
|
12 |
-
|
|
|
|
|
|
|
13 |
|
14 |
**Folder name details**:
|
15 |
- "add", "sub", or "mix": The types of questions the model can predict.
|
@@ -19,18 +28,4 @@ This repository contains a transformer model that can predict addition questions
|
|
19 |
- "t15K" to "t85K", etc.: The number of batches the model was trained on.
|
20 |
- "s372001", etc.: The random seed used in model training.
|
21 |
|
22 |
-
Some folder names also contain:
|
23 |
-
- "ins1": Before training, the model was initialized with a smaller, accurate addition model.
|
24 |
-
- "ins2": Same as ins1, but the inserted attention heads were not allowed to change.
|
25 |
-
- "ins3": Same as ins2, but the inserted MLP layers were also not allowed to change.
|
26 |
|
27 |
-
**Contents**:
|
28 |
-
- `model.pth`: The trained transformer model.
|
29 |
-
- `training_loss.json`: Data gathered during model training (used to plot "loss over training batches").
|
30 |
-
- `behaviors.json`: Facts gathered about the model by direct inspection (attention pattern data, PCA data, digit impact data, etc.).
|
31 |
-
- `features.json`: Facts gathered about hypothesized algorithm features via experimentation, e.g. node P12L0H1 implements the feature A3.ST.
|
32 |
-
|
33 |
-
**Provenance**:
|
34 |
-
- `model.pth` and `training_loss.json` were created by [QuantaMathsTrain.ipynb](https://github.com/PhilipQuirke/quanta_maths/blob/main/notebooks/QuantaMathsTrain.ipynb).
|
35 |
-
- `behaviors.json` and `features.json` were created by [QuantaMathsAnalyse.ipynb](https://github.com/PhilipQuirke/quanta_maths/blob/main/notebooks/QuantaMathsAnalyse.ipynb).
|
36 |
-
- The JSON files are used by [QuantaMathsAlgorithm.ipynb](https://github.com/PhilipQuirke/quanta_maths/blob/main/notebooks/QuantaMathsAlgorithm.ipynb).
|
|
|
1 |
# QuantaMaths: `add_d15_l2_h3_t80K_s572091`
|
2 |
|
3 |
+
This repository contains a transformer model that can predict addition questions.
|
4 |
|
5 |
+
### Model-specific metadata
|
6 |
+
- **Operation type**: addition
|
7 |
+
- **Num digits**: 15
|
8 |
+
- **Layers**: 2
|
9 |
+
- **Attention Heads**: 3
|
10 |
+
- **Training steps**: 80,000
|
11 |
+
- **Random seed**: 572091
|
12 |
+
**Contents**:
|
13 |
+
- `model.pth`: The trained transformer model.
|
14 |
+
- `training_loss.json`: Data gathered during model training (used to plot "loss over training batches").
|
15 |
+
- `behaviors.json`: Facts gathered about the model by direct inspection (attention pattern data, PCA data, digit impact data, etc.).
|
16 |
+
- `features.json`: Facts gathered about hypothesized algorithm features via experimentation, e.g. node P12L0H1 implements the feature A3.ST.
|
17 |
|
18 |
+
**Provenance**:
|
19 |
+
- `model.pth` and `training_loss.json` were created by [QuantaMathsTrain.ipynb](https://github.com/PhilipQuirke/quanta_maths/blob/main/notebooks/QuantaMathsTrain.ipynb).
|
20 |
+
- `behaviors.json` and `features.json` were created by [QuantaMathsAnalyse.ipynb](https://github.com/PhilipQuirke/quanta_maths/blob/main/notebooks/QuantaMathsAnalyse.ipynb).
|
21 |
+
- The JSON files are used by [QuantaMathsAlgorithm.ipynb](https://github.com/PhilipQuirke/quanta_maths/blob/main/notebooks/QuantaMathsAlgorithm.ipynb).
|
22 |
|
23 |
**Folder name details**:
|
24 |
- "add", "sub", or "mix": The types of questions the model can predict.
|
|
|
28 |
- "t15K" to "t85K", etc.: The number of batches the model was trained on.
|
29 |
- "s372001", etc.: The random seed used in model training.
|
30 |
|
|
|
|
|
|
|
|
|
31 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|