Davide Ghilardi
commited on
Commit
·
977dfba
1
Parent(s):
0dda12c
Add reverse l1_1
Browse files- l1_1/reversed/attn_out_layer0/cfg.json +86 -0
- l1_1/reversed/attn_out_layer0/sae_weights.safetensors +3 -0
- l1_1/reversed/attn_out_layer0/sparsity.safetensors +3 -0
- l1_1/reversed/mlp_out_layer0/cfg.json +1 -0
- l1_1/reversed/mlp_out_layer0/sae_weights.safetensors +3 -0
- l1_1/reversed/mlp_out_layer0/sparsity.safetensors +3 -0
- l1_1/reversed/resid_out_layer0/cfg.json +1 -0
- l1_1/reversed/resid_out_layer0/sae_weights.safetensors +3 -0
- l1_1/reversed/resid_out_layer0/sparsity.safetensors +3 -0
l1_1/reversed/attn_out_layer0/cfg.json
ADDED
|
@@ -0,0 +1,86 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"model_name": "1l-gelu",
|
| 3 |
+
"model_class_name": "HookedTransformer",
|
| 4 |
+
"hook_name": "blocks.0.attn.hook_z",
|
| 5 |
+
"hook_eval": "NOT_IN_USE",
|
| 6 |
+
"hook_layer": 0,
|
| 7 |
+
"hook_head_index": null,
|
| 8 |
+
"dataset_path": "ghidav/arithmetics_reversed",
|
| 9 |
+
"dataset_trust_remote_code": true,
|
| 10 |
+
"streaming": true,
|
| 11 |
+
"is_dataset_tokenized": true,
|
| 12 |
+
"context_size": 18,
|
| 13 |
+
"use_cached_activations": false,
|
| 14 |
+
"cached_activations_path": null,
|
| 15 |
+
"architecture": "jumprelu",
|
| 16 |
+
"d_in": 512,
|
| 17 |
+
"d_sae": 8192,
|
| 18 |
+
"b_dec_init_method": "zeros",
|
| 19 |
+
"expansion_factor": 16,
|
| 20 |
+
"activation_fn": "relu",
|
| 21 |
+
"activation_fn_kwargs": {},
|
| 22 |
+
"normalize_sae_decoder": true,
|
| 23 |
+
"noise_scale": 0.0,
|
| 24 |
+
"from_pretrained_path": null,
|
| 25 |
+
"apply_b_dec_to_input": false,
|
| 26 |
+
"decoder_orthogonal_init": false,
|
| 27 |
+
"decoder_heuristic_init": false,
|
| 28 |
+
"init_encoder_as_decoder_transpose": false,
|
| 29 |
+
"n_batches_in_buffer": 128,
|
| 30 |
+
"training_tokens": 50000000,
|
| 31 |
+
"finetuning_tokens": 0,
|
| 32 |
+
"store_batch_size_prompts": 8,
|
| 33 |
+
"train_batch_size_tokens": 1024,
|
| 34 |
+
"normalize_activations": "none",
|
| 35 |
+
"device": "cuda",
|
| 36 |
+
"act_store_device": "cuda",
|
| 37 |
+
"seed": 42,
|
| 38 |
+
"dtype": "float32",
|
| 39 |
+
"prepend_bos": false,
|
| 40 |
+
"autocast": false,
|
| 41 |
+
"autocast_lm": false,
|
| 42 |
+
"compile_llm": false,
|
| 43 |
+
"llm_compilation_mode": null,
|
| 44 |
+
"compile_sae": false,
|
| 45 |
+
"sae_compilation_mode": null,
|
| 46 |
+
"adam_beta1": 0,
|
| 47 |
+
"adam_beta2": 0.999,
|
| 48 |
+
"mse_loss_normalization": null,
|
| 49 |
+
"l1_coefficient": 1.0,
|
| 50 |
+
"lp_norm": 1,
|
| 51 |
+
"scale_sparsity_penalty_by_decoder_norm": false,
|
| 52 |
+
"l1_warm_up_steps": 2441,
|
| 53 |
+
"lr": 0.0005,
|
| 54 |
+
"lr_scheduler_name": "constant",
|
| 55 |
+
"lr_warm_up_steps": 0,
|
| 56 |
+
"lr_end": 5e-05,
|
| 57 |
+
"lr_decay_steps": 9765,
|
| 58 |
+
"n_restart_cycles": 1,
|
| 59 |
+
"finetuning_method": null,
|
| 60 |
+
"use_ghost_grads": false,
|
| 61 |
+
"feature_sampling_window": 2000,
|
| 62 |
+
"dead_feature_window": 1000,
|
| 63 |
+
"dead_feature_threshold": 1e-06,
|
| 64 |
+
"n_eval_batches": 10,
|
| 65 |
+
"eval_batch_size_prompts": null,
|
| 66 |
+
"log_to_wandb": true,
|
| 67 |
+
"log_activations_store_to_wandb": false,
|
| 68 |
+
"log_optimizer_state_to_wandb": false,
|
| 69 |
+
"wandb_project": "sae-feature-circuits",
|
| 70 |
+
"wandb_id": null,
|
| 71 |
+
"run_name": "L0_attn.hook_z_L1_1_0_rev",
|
| 72 |
+
"wandb_entity": null,
|
| 73 |
+
"wandb_log_frequency": 30,
|
| 74 |
+
"eval_every_n_wandb_logs": 100,
|
| 75 |
+
"resume": false,
|
| 76 |
+
"n_checkpoints": 0,
|
| 77 |
+
"checkpoint_path": "checkpoints/ksq6xya7",
|
| 78 |
+
"verbose": false,
|
| 79 |
+
"model_kwargs": {},
|
| 80 |
+
"model_from_pretrained_kwargs": {
|
| 81 |
+
"center_writing_weights": false
|
| 82 |
+
},
|
| 83 |
+
"sae_lens_version": "3.20.5",
|
| 84 |
+
"sae_lens_training_version": "3.20.5",
|
| 85 |
+
"tokens_per_buffer": 2359296
|
| 86 |
+
}
|
l1_1/reversed/attn_out_layer0/sae_weights.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:53f05bef3182877f0ca82ea045fc120240b98d0a23313ff7040c5d414fc71464
|
| 3 |
+
size 33622400
|
l1_1/reversed/attn_out_layer0/sparsity.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:060a8e2c43f1d1c6254169289a7d732150319f08f747b9b141b39f70557fbea7
|
| 3 |
+
size 32848
|
l1_1/reversed/mlp_out_layer0/cfg.json
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
{"model_name": "1l-gelu", "model_class_name": "HookedTransformer", "hook_name": "blocks.0.hook_mlp_out", "hook_eval": "NOT_IN_USE", "hook_layer": 0, "hook_head_index": null, "dataset_path": "ghidav/arithmetics_reversed", "dataset_trust_remote_code": true, "streaming": true, "is_dataset_tokenized": true, "context_size": 18, "use_cached_activations": false, "cached_activations_path": null, "architecture": "jumprelu", "d_in": 512, "d_sae": 8192, "b_dec_init_method": "zeros", "expansion_factor": 16, "activation_fn": "relu", "activation_fn_kwargs": {}, "normalize_sae_decoder": true, "noise_scale": 0.0, "from_pretrained_path": null, "apply_b_dec_to_input": false, "decoder_orthogonal_init": false, "decoder_heuristic_init": false, "init_encoder_as_decoder_transpose": false, "n_batches_in_buffer": 128, "training_tokens": 50000000, "finetuning_tokens": 0, "store_batch_size_prompts": 8, "train_batch_size_tokens": 1024, "normalize_activations": "none", "device": "cuda", "act_store_device": "cuda", "seed": 42, "dtype": "float32", "prepend_bos": false, "autocast": false, "autocast_lm": false, "compile_llm": false, "llm_compilation_mode": null, "compile_sae": false, "sae_compilation_mode": null, "adam_beta1": 0, "adam_beta2": 0.999, "mse_loss_normalization": null, "l1_coefficient": 1.0, "lp_norm": 1, "scale_sparsity_penalty_by_decoder_norm": false, "l1_warm_up_steps": 2441, "lr": 0.0005, "lr_scheduler_name": "constant", "lr_warm_up_steps": 0, "lr_end": 5e-05, "lr_decay_steps": 9765, "n_restart_cycles": 1, "finetuning_method": null, "use_ghost_grads": false, "feature_sampling_window": 2000, "dead_feature_window": 1000, "dead_feature_threshold": 1e-06, "n_eval_batches": 10, "eval_batch_size_prompts": null, "log_to_wandb": true, "log_activations_store_to_wandb": false, "log_optimizer_state_to_wandb": false, "wandb_project": "sae-feature-circuits", "wandb_id": null, "run_name": "L0_hook_mlp_out_L1_1_0_rev", "wandb_entity": null, "wandb_log_frequency": 30, "eval_every_n_wandb_logs": 100, "resume": false, "n_checkpoints": 0, "checkpoint_path": "checkpoints/b901vcmz", "verbose": false, "model_kwargs": {}, "model_from_pretrained_kwargs": {"center_writing_weights": false}, "sae_lens_version": "3.20.5", "sae_lens_training_version": "3.20.5", "tokens_per_buffer": 2359296}
|
l1_1/reversed/mlp_out_layer0/sae_weights.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7a1b85fef7ce03ef1222cc7f3a1861176b4566997237cbbdee96651bb5075239
|
| 3 |
+
size 33622400
|
l1_1/reversed/mlp_out_layer0/sparsity.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c58e029f81b4a19381d4e6ce6efd8524d8dea5d89826983b0b63ab39d3c7c8a1
|
| 3 |
+
size 32848
|
l1_1/reversed/resid_out_layer0/cfg.json
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
{"model_name": "1l-gelu", "model_class_name": "HookedTransformer", "hook_name": "blocks.0.hook_resid_post", "hook_eval": "NOT_IN_USE", "hook_layer": 0, "hook_head_index": null, "dataset_path": "ghidav/arithmetics_reversed", "dataset_trust_remote_code": true, "streaming": true, "is_dataset_tokenized": true, "context_size": 18, "use_cached_activations": false, "cached_activations_path": null, "architecture": "jumprelu", "d_in": 512, "d_sae": 8192, "b_dec_init_method": "zeros", "expansion_factor": 16, "activation_fn": "relu", "activation_fn_kwargs": {}, "normalize_sae_decoder": true, "noise_scale": 0.0, "from_pretrained_path": null, "apply_b_dec_to_input": false, "decoder_orthogonal_init": false, "decoder_heuristic_init": false, "init_encoder_as_decoder_transpose": false, "n_batches_in_buffer": 128, "training_tokens": 50000000, "finetuning_tokens": 0, "store_batch_size_prompts": 8, "train_batch_size_tokens": 1024, "normalize_activations": "none", "device": "cuda", "act_store_device": "cuda", "seed": 42, "dtype": "float32", "prepend_bos": false, "autocast": false, "autocast_lm": false, "compile_llm": false, "llm_compilation_mode": null, "compile_sae": false, "sae_compilation_mode": null, "adam_beta1": 0, "adam_beta2": 0.999, "mse_loss_normalization": null, "l1_coefficient": 1.0, "lp_norm": 1, "scale_sparsity_penalty_by_decoder_norm": false, "l1_warm_up_steps": 2441, "lr": 0.0005, "lr_scheduler_name": "constant", "lr_warm_up_steps": 0, "lr_end": 5e-05, "lr_decay_steps": 9765, "n_restart_cycles": 1, "finetuning_method": null, "use_ghost_grads": false, "feature_sampling_window": 2000, "dead_feature_window": 1000, "dead_feature_threshold": 1e-06, "n_eval_batches": 10, "eval_batch_size_prompts": null, "log_to_wandb": true, "log_activations_store_to_wandb": false, "log_optimizer_state_to_wandb": false, "wandb_project": "sae-feature-circuits", "wandb_id": null, "run_name": "L0_hook_resid_post_L1_1_0_rev", "wandb_entity": null, "wandb_log_frequency": 30, "eval_every_n_wandb_logs": 100, "resume": false, "n_checkpoints": 0, "checkpoint_path": "checkpoints/fk64hm4k", "verbose": false, "model_kwargs": {}, "model_from_pretrained_kwargs": {"center_writing_weights": false}, "sae_lens_version": "3.20.5", "sae_lens_training_version": "3.20.5", "tokens_per_buffer": 2359296}
|
l1_1/reversed/resid_out_layer0/sae_weights.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:1c6fb090b0ed1bd35f51112a050b86b479917aff5a047402ecdcf493070b37b9
|
| 3 |
+
size 33622400
|
l1_1/reversed/resid_out_layer0/sparsity.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:48024286c1aa07a8e7f2c9d1d1e35abebd6f81754a0ae8e0cc0400de00b33199
|
| 3 |
+
size 32848
|