Card wording; drop unused cell_id from probe metadata
Browse files- README.md +2 -2
- nemotron-3-super-120b/probes/axial/probe_metadata.json +0 -1
- nemotron-3-super-120b/probes/efc/probe_metadata.json +0 -1
- qwen3.5-9b/probes/axial/probe_metadata.json +0 -1
- qwen3.5-9b/probes/efc/probe_metadata.json +0 -1
- qwen3.5-9b/probes/linear/probe_metadata.json +0 -1
- qwen3.5-9b/probes/mlp/probe_metadata.json +0 -1
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
|
| 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>/`:
|
| 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,
|