chrisjcundy commited on
Commit
9f107dc
·
verified ·
1 Parent(s): 322a8aa

Card wording; drop unused cell_id from probe metadata

Browse files
README.md CHANGED
@@ -11,7 +11,7 @@ Test fixture for the `probe-inference` package's parity tests (`tests/test_parit
11
  download this dataset at a pinned revision and check that the package reproduces reference scores on
12
  real activations.
13
 
14
- For each model it holds a few rows of residual-stream activations and token masks, the probe cells, and
15
  the reference score of each probe on each row:
16
 
17
  | Directory | Model | Rows | Probes |
@@ -24,7 +24,7 @@ Files in each model directory:
24
  - `acts.pt`: `{layer: Tensor[rows, seq, d_model]}` in bfloat16, the outputs of decoder block `layer`
25
  (Hugging Face `hidden_states[layer + 1]`), over each row's follow-up window.
26
  - `masks.pt`: `tokens`, `prompt_mask`, `completion_mask` and `followup_start_positions` for the same rows.
27
- - `probes/<arch>/`: probe cells in the layout `probe_inference.load_probe` reads.
28
  - `reference.json`: per probe, `score` (EFC, axial), or `L<layer>` logits plus `combined` (the mean of
29
  per-layer sigmoids) for linear and MLP, one value per row.
30
 
 
11
  download this dataset at a pinned revision and check that the package reproduces reference scores on
12
  real activations.
13
 
14
+ For each model it holds a few rows of residual-stream activations and token masks, the trained probes, and
15
  the reference score of each probe on each row:
16
 
17
  | Directory | Model | Rows | Probes |
 
24
  - `acts.pt`: `{layer: Tensor[rows, seq, d_model]}` in bfloat16, the outputs of decoder block `layer`
25
  (Hugging Face `hidden_states[layer + 1]`), over each row's follow-up window.
26
  - `masks.pt`: `tokens`, `prompt_mask`, `completion_mask` and `followup_start_positions` for the same rows.
27
+ - `probes/<arch>/`: trained probes in the directory layout `probe_inference.load_probe` reads.
28
  - `reference.json`: per probe, `score` (EFC, axial), or `L<layer>` logits plus `combined` (the mean of
29
  per-layer sigmoids) for linear and MLP, one value per row.
30
 
nemotron-3-super-120b/probes/axial/probe_metadata.json CHANGED
@@ -1,5 +1,4 @@
1
  {
2
- "cell_id": "nemotron_super_axial",
3
  "layers": [
4
  26,
5
  37,
 
1
  {
 
2
  "layers": [
3
  26,
4
  37,
nemotron-3-super-120b/probes/efc/probe_metadata.json CHANGED
@@ -1,5 +1,4 @@
1
  {
2
- "cell_id": "nemotron_super_efc",
3
  "layers": [
4
  26,
5
  37,
 
1
  {
 
2
  "layers": [
3
  26,
4
  37,
qwen3.5-9b/probes/axial/probe_metadata.json CHANGED
@@ -1,5 +1,4 @@
1
  {
2
- "cell_id": "qwen9b_axial",
3
  "layers": [
4
  10,
5
  13,
 
1
  {
 
2
  "layers": [
3
  10,
4
  13,
qwen3.5-9b/probes/efc/probe_metadata.json CHANGED
@@ -1,5 +1,4 @@
1
  {
2
- "cell_id": "qwen9b_efc",
3
  "layers": [
4
  10,
5
  13,
 
1
  {
 
2
  "layers": [
3
  10,
4
  13,
qwen3.5-9b/probes/linear/probe_metadata.json CHANGED
@@ -1,5 +1,4 @@
1
  {
2
- "cell_id": "qwen9b_linear",
3
  "layers": [
4
  10,
5
  13,
 
1
  {
 
2
  "layers": [
3
  10,
4
  13,
qwen3.5-9b/probes/mlp/probe_metadata.json CHANGED
@@ -1,5 +1,4 @@
1
  {
2
- "cell_id": "qwen9b_mlp",
3
  "layers": [
4
  10,
5
  13,
 
1
  {
 
2
  "layers": [
3
  10,
4
  13,