chrisjcundy commited on
Commit
662e1da
·
verified ·
1 Parent(s): b8131eb

Card: describe each directory as a trained probe

Browse files
Files changed (1) hide show
  1. README.md +5 -5
README.md CHANGED
@@ -17,15 +17,15 @@ layers and returns one score per transcript. Load them with the `probe-inference
17
  ```python
18
  from probe_inference import load_probe_from_hub
19
 
20
- bank = load_probe_from_hub("qwen3.5-9b/efc") # this repository at the package's pinned revision
21
- score = bank.score(acts, bank.read_mask(prompt_mask, completion_mask, followup_start_position))
22
  ```
23
 
24
  ## Layout
25
 
26
- `<model>/<arch>/`, with `<arch>` in `linear`, `mlp`, `efc`, `axial`. A linear or MLP cell holds one probe per
27
- layer (`layer_<L>/config.json`, `layer_<L>/model.pt`); an EFC or axial cell holds one cross-layer probe
28
- (`config.json`, `model.pt`). Every cell has `probe_metadata.json`: the model and revision, the layers, the
29
  read window (`obfuscate_over`), the token aggregation and, for linear and MLP, the layers whose sigmoids
30
  are averaged (`layer_rule.used_layers`). `model.pt` files are plain float32 state dicts.
31
 
 
17
  ```python
18
  from probe_inference import load_probe_from_hub
19
 
20
+ probe = load_probe_from_hub("qwen3.5-9b/efc") # this repository at the package's pinned revision
21
+ score = probe.score(acts, probe.read_mask(prompt_mask, completion_mask, followup_start_position))
22
  ```
23
 
24
  ## Layout
25
 
26
+ `<model>/<arch>/`, with `<arch>` in `linear`, `mlp`, `efc`, `axial`. Each directory is one trained probe. A linear or MLP probe is one
27
+ small probe per layer (`layer_<L>/config.json`, `layer_<L>/model.pt`); an EFC or axial probe is one module
28
+ that reads all its layers at once (`config.json`, `model.pt`). Every probe has `probe_metadata.json`: the model and revision, the layers, the
29
  read window (`obfuscate_over`), the token aggregation and, for linear and MLP, the layers whose sigmoids
30
  are averaged (`layer_rule.used_layers`). `model.pt` files are plain float32 state dicts.
31