File size: 716 Bytes
f90603a c843477 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
---
license: cc-by-4.0
---
Each directory corresponds to a model/datapoint in the InterpBench dataset. It is structured as:
```
- task // directory name
-- ll_model.pth // the low level transformer model
-- ll_model_cfg.pkl // a config for the transformer model
-- meta.json // training hyperparams
-- edges.pkl // label for the circuit, i.e., list of all the edges that are a part of the ground truth circuit
```
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 |