cybershiptrooper commited on
Commit
c843477
1 Parent(s): f7c1d5b

update readme

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