cybershiptrooper
commited on
Commit
•
c843477
1
Parent(s):
f7c1d5b
update readme
Browse files
README.md
CHANGED
@@ -1,3 +1,15 @@
|
|
1 |
---
|
2 |
license: apache-2.0
|
3 |
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
---
|
2 |
license: apache-2.0
|
3 |
---
|
4 |
+
|
5 |
+
Each directory corresponds to a model/datapoint in the InterpBench dataset. It is structured as:
|
6 |
+
|
7 |
+
```
|
8 |
+
- task // directory name
|
9 |
+
-- ll_model_{weight}.pth // the low level transformer model
|
10 |
+
-- meta.json //ll_model_cfg_{weight}.pkl // a config for the transformer model
|
11 |
+
-- meta_{weight}.json // training hyperparams
|
12 |
+
-- edges.pkl // label for the circuit, i.e., list of all the edges that are a part of the ground truth circuit
|
13 |
+
```
|
14 |
+
|
15 |
+
This repository of models is complimentary to [CircuitsBenchmark](https://github.com/FlyingPumba/circuits-benchmark), and should be used to load the models. Alternatively, [TransformerLens](https://github.com/TransformerLensOrg/TransformerLens) can also be used to load it using the ll_config.json
|