adamkarvonen commited on
Commit
cc7515c
·
verified ·
1 Parent(s): ad02de2

Upload folder using huggingface_hub

Browse files
._saes_Qwen_Qwen2.5-Coder-32B-Instruct_batch_top_k/resid_post_layer_50/trainer_0/ae.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:651bc22061b540e3fbe97f330460663104ff69c61d5600c6664fd5b0a7798a80
3
+ size 671176726
._saes_Qwen_Qwen2.5-Coder-32B-Instruct_batch_top_k/resid_post_layer_50/trainer_0/config.json ADDED
@@ -0,0 +1,32 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "trainer": {
3
+ "trainer_class": "BatchTopKTrainer",
4
+ "dict_class": "BatchTopKSAE",
5
+ "lr": 0.0003,
6
+ "steps": 122070,
7
+ "auxk_alpha": 0.03125,
8
+ "warmup_steps": 1000,
9
+ "decay_start": 97656,
10
+ "threshold_beta": 0.999,
11
+ "threshold_start_step": 1000,
12
+ "top_k_aux": 2560,
13
+ "seed": 0,
14
+ "activation_dim": 5120,
15
+ "dict_size": 16384,
16
+ "k": 80,
17
+ "device": "cuda:0",
18
+ "layer": 50,
19
+ "lm_name": "Qwen/Qwen2.5-Coder-32B-Instruct",
20
+ "wandb_name": "BatchTopKTrainer-Qwen/Qwen2.5-Coder-32B-Instruct-resid_post_layer_50_trainer_0",
21
+ "submodule_name": "resid_post_layer_50"
22
+ },
23
+ "buffer": {
24
+ "d_submodule": 5120,
25
+ "io": "out",
26
+ "n_ctxs": 122,
27
+ "ctx_len": 2048,
28
+ "refresh_batch_size": 4,
29
+ "out_batch_size": 4096,
30
+ "device": "cuda:0"
31
+ }
32
+ }
._saes_Qwen_Qwen2.5-Coder-32B-Instruct_batch_top_k/resid_post_layer_50/trainer_0/eval_results.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"l2_loss": 232.7, "l1_loss": 2156.88, "l0": 78.24118896484374, "frac_variance_explained": 0.88583984375, "cossim": 0.871328125, "l2_ratio": 0.8742578125, "relative_reconstruction_bias": 0.9945703125, "frac_alive": 0.94476318359375, "hyperparameters": {"n_inputs": 200, "context_length": 2048}}
._saes_Qwen_Qwen2.5-Coder-32B-Instruct_batch_top_k/resid_post_layer_50/trainer_1/ae.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5d169b1d707ff009a1dca853096135308df0edf299ceafa16147d9c98a2e64bf
3
+ size 671176726
._saes_Qwen_Qwen2.5-Coder-32B-Instruct_batch_top_k/resid_post_layer_50/trainer_1/config.json ADDED
@@ -0,0 +1,32 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "trainer": {
3
+ "trainer_class": "BatchTopKTrainer",
4
+ "dict_class": "BatchTopKSAE",
5
+ "lr": 0.0003,
6
+ "steps": 122070,
7
+ "auxk_alpha": 0.03125,
8
+ "warmup_steps": 1000,
9
+ "decay_start": 97656,
10
+ "threshold_beta": 0.999,
11
+ "threshold_start_step": 1000,
12
+ "top_k_aux": 2560,
13
+ "seed": 0,
14
+ "activation_dim": 5120,
15
+ "dict_size": 16384,
16
+ "k": 160,
17
+ "device": "cuda:0",
18
+ "layer": 50,
19
+ "lm_name": "Qwen/Qwen2.5-Coder-32B-Instruct",
20
+ "wandb_name": "BatchTopKTrainer-Qwen/Qwen2.5-Coder-32B-Instruct-resid_post_layer_50_trainer_1",
21
+ "submodule_name": "resid_post_layer_50"
22
+ },
23
+ "buffer": {
24
+ "d_submodule": 5120,
25
+ "io": "out",
26
+ "n_ctxs": 122,
27
+ "ctx_len": 2048,
28
+ "refresh_batch_size": 4,
29
+ "out_batch_size": 4096,
30
+ "device": "cuda:0"
31
+ }
32
+ }
._saes_Qwen_Qwen2.5-Coder-32B-Instruct_batch_top_k/resid_post_layer_50/trainer_1/eval_results.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"l2_loss": 212.12, "l1_loss": 3819.52, "l0": 161.6133349609375, "frac_variance_explained": 0.9108984375, "cossim": 0.894375, "l2_ratio": 0.896171875, "relative_reconstruction_bias": 0.991875, "frac_alive": 0.9765625, "hyperparameters": {"n_inputs": 200, "context_length": 2048}}
._saes_Qwen_Qwen2.5-Coder-32B-Instruct_batch_top_k/resid_post_layer_50/trainer_2/ae.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c666e63996352b26b4064ae89b9e48fed55001e63af13094ee2b12c8940b5eb4
3
+ size 2684639254
._saes_Qwen_Qwen2.5-Coder-32B-Instruct_batch_top_k/resid_post_layer_50/trainer_2/config.json ADDED
@@ -0,0 +1,32 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "trainer": {
3
+ "trainer_class": "BatchTopKTrainer",
4
+ "dict_class": "BatchTopKSAE",
5
+ "lr": 0.0003,
6
+ "steps": 122070,
7
+ "auxk_alpha": 0.03125,
8
+ "warmup_steps": 1000,
9
+ "decay_start": 97656,
10
+ "threshold_beta": 0.999,
11
+ "threshold_start_step": 1000,
12
+ "top_k_aux": 2560,
13
+ "seed": 0,
14
+ "activation_dim": 5120,
15
+ "dict_size": 65536,
16
+ "k": 80,
17
+ "device": "cuda:0",
18
+ "layer": 50,
19
+ "lm_name": "Qwen/Qwen2.5-Coder-32B-Instruct",
20
+ "wandb_name": "BatchTopKTrainer-Qwen/Qwen2.5-Coder-32B-Instruct-resid_post_layer_50_trainer_2",
21
+ "submodule_name": "resid_post_layer_50"
22
+ },
23
+ "buffer": {
24
+ "d_submodule": 5120,
25
+ "io": "out",
26
+ "n_ctxs": 122,
27
+ "ctx_len": 2048,
28
+ "refresh_batch_size": 4,
29
+ "out_batch_size": 4096,
30
+ "device": "cuda:0"
31
+ }
32
+ }
._saes_Qwen_Qwen2.5-Coder-32B-Instruct_batch_top_k/resid_post_layer_50/trainer_2/eval_results.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"l2_loss": 219.34, "l1_loss": 2060.84, "l0": 78.80491455078125, "frac_variance_explained": 0.90408203125, "cossim": 0.88654296875, "l2_ratio": 0.8886328125, "relative_reconstruction_bias": 0.99619140625, "frac_alive": 0.7082977294921875, "hyperparameters": {"n_inputs": 200, "context_length": 2048}}
._saes_Qwen_Qwen2.5-Coder-32B-Instruct_batch_top_k/resid_post_layer_50/trainer_3/ae.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5abdeebc0775abe86d2522fcf15cc533b44ee960f0c6f7ae367cdbcd19bc10a2
3
+ size 2684639254
._saes_Qwen_Qwen2.5-Coder-32B-Instruct_batch_top_k/resid_post_layer_50/trainer_3/config.json ADDED
@@ -0,0 +1,32 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "trainer": {
3
+ "trainer_class": "BatchTopKTrainer",
4
+ "dict_class": "BatchTopKSAE",
5
+ "lr": 0.0003,
6
+ "steps": 122070,
7
+ "auxk_alpha": 0.03125,
8
+ "warmup_steps": 1000,
9
+ "decay_start": 97656,
10
+ "threshold_beta": 0.999,
11
+ "threshold_start_step": 1000,
12
+ "top_k_aux": 2560,
13
+ "seed": 0,
14
+ "activation_dim": 5120,
15
+ "dict_size": 65536,
16
+ "k": 160,
17
+ "device": "cuda:0",
18
+ "layer": 50,
19
+ "lm_name": "Qwen/Qwen2.5-Coder-32B-Instruct",
20
+ "wandb_name": "BatchTopKTrainer-Qwen/Qwen2.5-Coder-32B-Instruct-resid_post_layer_50_trainer_3",
21
+ "submodule_name": "resid_post_layer_50"
22
+ },
23
+ "buffer": {
24
+ "d_submodule": 5120,
25
+ "io": "out",
26
+ "n_ctxs": 122,
27
+ "ctx_len": 2048,
28
+ "refresh_batch_size": 4,
29
+ "out_batch_size": 4096,
30
+ "device": "cuda:0"
31
+ }
32
+ }
._saes_Qwen_Qwen2.5-Coder-32B-Instruct_batch_top_k/resid_post_layer_50/trainer_3/eval_results.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"l2_loss": 198.33, "l1_loss": 3952.96, "l0": 161.882958984375, "frac_variance_explained": 0.9212109375, "cossim": 0.90740234375, "l2_ratio": 0.90974609375, "relative_reconstruction_bias": 0.995078125, "frac_alive": 0.734619140625, "hyperparameters": {"n_inputs": 200, "context_length": 2048}}