PhilipQuirke commited on
Commit
0dd04fb
·
verified ·
1 Parent(s): b54e2b9

Update model card with revised readme

Browse files
Files changed (1) hide show
  1. README.md +17 -22
README.md CHANGED
@@ -1,15 +1,24 @@
1
  # QuantaMaths: `mix_d11_l3_h4_t80K_s572091`
2
 
3
- ### Model-specific metadata for `mix_d11_l3_h4_t80K_s572091`
4
 
5
- - **Operation type**: mix
6
- - **Max digits**: d11
7
- - **Layers**: l3
8
- - **Attention Heads**: h4
9
- - **Training steps**: t80K
10
- - **Random seed**: s572091
 
 
 
 
 
 
11
 
12
- This repository contains a transformer model that can predict addition questions, subtraction questions, or both.
 
 
 
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: `mix_d11_l3_h4_t80K_s572091`
2
 
3
+ This repository contains a transformer model that can predict both addition and subtraction questions.
4
 
5
+ ### Model-specific metadata
6
+ - **Operation type**: mixed
7
+ - **Num digits**: 11
8
+ - **Layers**: 3
9
+ - **Attention Heads**: 4
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