Upload folder using huggingface_hub
Browse files- ._test_run_EleutherAI_pythia-70m-deduped_top_k_sampled_sae/resid_post_layer_3/SampledActivationTrainer-entropy-EleutherAI_pythia-70m-deduped-resid_post_layer_3_trainer_2/ae.pt +3 -0
- ._test_run_EleutherAI_pythia-70m-deduped_top_k_sampled_sae/resid_post_layer_3/SampledActivationTrainer-entropy-EleutherAI_pythia-70m-deduped-resid_post_layer_3_trainer_2/config.json +36 -0
- ._test_run_EleutherAI_pythia-70m-deduped_top_k_sampled_sae/resid_post_layer_3/SampledActivationTrainer-entropy-EleutherAI_pythia-70m-deduped-resid_post_layer_3_trainer_2/eval_results.json +1 -0
- ._test_run_EleutherAI_pythia-70m-deduped_top_k_sampled_sae/resid_post_layer_3/SampledActivationTrainer-leverage-EleutherAI_pythia-70m-deduped-resid_post_layer_3_trainer_1/ae.pt +3 -0
- ._test_run_EleutherAI_pythia-70m-deduped_top_k_sampled_sae/resid_post_layer_3/SampledActivationTrainer-leverage-EleutherAI_pythia-70m-deduped-resid_post_layer_3_trainer_1/config.json +36 -0
- ._test_run_EleutherAI_pythia-70m-deduped_top_k_sampled_sae/resid_post_layer_3/SampledActivationTrainer-leverage-EleutherAI_pythia-70m-deduped-resid_post_layer_3_trainer_1/eval_results.json +1 -0
- ._test_run_EleutherAI_pythia-70m-deduped_top_k_sampled_sae/resid_post_layer_3/TopKTrainer-EleutherAI_pythia-70m-deduped-resid_post_layer_3_trainer_0/ae.pt +3 -0
- ._test_run_EleutherAI_pythia-70m-deduped_top_k_sampled_sae/resid_post_layer_3/TopKTrainer-EleutherAI_pythia-70m-deduped-resid_post_layer_3_trainer_0/config.json +31 -0
- ._test_run_EleutherAI_pythia-70m-deduped_top_k_sampled_sae/resid_post_layer_3/TopKTrainer-EleutherAI_pythia-70m-deduped-resid_post_layer_3_trainer_0/eval_results.json +1 -0
._test_run_EleutherAI_pythia-70m-deduped_top_k_sampled_sae/resid_post_layer_3/SampledActivationTrainer-entropy-EleutherAI_pythia-70m-deduped-resid_post_layer_3_trainer_2/ae.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7a663a086b7460f06df1f8560ac2df2489bdcd4a97b4260524b03fbece3238d5
|
| 3 |
+
size 268964962
|
._test_run_EleutherAI_pythia-70m-deduped_top_k_sampled_sae/resid_post_layer_3/SampledActivationTrainer-entropy-EleutherAI_pythia-70m-deduped-resid_post_layer_3_trainer_2/config.json
ADDED
|
@@ -0,0 +1,36 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"trainer": {
|
| 3 |
+
"trainer_class": "SampledActivationTrainer",
|
| 4 |
+
"dict_class": "SampledActivationSAE",
|
| 5 |
+
"lr": 0.0003,
|
| 6 |
+
"steps": 48,
|
| 7 |
+
"auxk_alpha": 0.03125,
|
| 8 |
+
"warmup_steps": 1000,
|
| 9 |
+
"decay_start": 38,
|
| 10 |
+
"threshold_beta": 0.999,
|
| 11 |
+
"threshold_start_step": 1000,
|
| 12 |
+
"sampling_update_freq": 1,
|
| 13 |
+
"sampling_method": "entropy",
|
| 14 |
+
"ridge_lambda": 0.01,
|
| 15 |
+
"sketching_size": 100,
|
| 16 |
+
"top_k_aux": 256,
|
| 17 |
+
"seed": 0,
|
| 18 |
+
"activation_dim": 512,
|
| 19 |
+
"dict_size": 65536,
|
| 20 |
+
"k": 20,
|
| 21 |
+
"device": "cuda:0",
|
| 22 |
+
"layer": 3,
|
| 23 |
+
"lm_name": "EleutherAI/pythia-70m-deduped",
|
| 24 |
+
"wandb_name": "SampledActivationTrainer-entropy-EleutherAI/pythia-70m-deduped-resid_post_layer_3_trainer_2",
|
| 25 |
+
"submodule_name": "resid_post_layer_3"
|
| 26 |
+
},
|
| 27 |
+
"buffer": {
|
| 28 |
+
"d_submodule": 512,
|
| 29 |
+
"io": "out",
|
| 30 |
+
"n_ctxs": 244,
|
| 31 |
+
"ctx_len": 1024,
|
| 32 |
+
"refresh_batch_size": 64,
|
| 33 |
+
"out_batch_size": 2048,
|
| 34 |
+
"device": "cuda:0"
|
| 35 |
+
}
|
| 36 |
+
}
|
._test_run_EleutherAI_pythia-70m-deduped_top_k_sampled_sae/resid_post_layer_3/SampledActivationTrainer-entropy-EleutherAI_pythia-70m-deduped-resid_post_layer_3_trainer_2/eval_results.json
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
{"l2_loss": 767.1420555114746, "l1_loss": 14069.71826171875, "l0": 32533.035400390625, "frac_variance_explained": -3754.4764709472656, "cossim": 0.8404335379600525, "l2_ratio": 53.196314573287964, "relative_reconstruction_bias": 62.36626052856445, "loss_original": 2.997953861951828, "loss_reconstructed": 18.35732865333557, "loss_zero": 12.705852329730988, "frac_recovered": -0.5863620117306709, "frac_alive": 1.0, "hyperparameters": {"n_inputs": 200, "context_length": 1024}}
|
._test_run_EleutherAI_pythia-70m-deduped_top_k_sampled_sae/resid_post_layer_3/SampledActivationTrainer-leverage-EleutherAI_pythia-70m-deduped-resid_post_layer_3_trainer_1/ae.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7abec87c5a8d6e0c50e00652a056c7d172d1748ea2682e1a206d92db9b6a49ad
|
| 3 |
+
size 268964962
|
._test_run_EleutherAI_pythia-70m-deduped_top_k_sampled_sae/resid_post_layer_3/SampledActivationTrainer-leverage-EleutherAI_pythia-70m-deduped-resid_post_layer_3_trainer_1/config.json
ADDED
|
@@ -0,0 +1,36 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"trainer": {
|
| 3 |
+
"trainer_class": "SampledActivationTrainer",
|
| 4 |
+
"dict_class": "SampledActivationSAE",
|
| 5 |
+
"lr": 0.0003,
|
| 6 |
+
"steps": 48,
|
| 7 |
+
"auxk_alpha": 0.03125,
|
| 8 |
+
"warmup_steps": 1000,
|
| 9 |
+
"decay_start": 38,
|
| 10 |
+
"threshold_beta": 0.999,
|
| 11 |
+
"threshold_start_step": 1000,
|
| 12 |
+
"sampling_update_freq": 1,
|
| 13 |
+
"sampling_method": "leverage",
|
| 14 |
+
"ridge_lambda": 0.01,
|
| 15 |
+
"sketching_size": 100,
|
| 16 |
+
"top_k_aux": 256,
|
| 17 |
+
"seed": 0,
|
| 18 |
+
"activation_dim": 512,
|
| 19 |
+
"dict_size": 65536,
|
| 20 |
+
"k": 20,
|
| 21 |
+
"device": "cuda:0",
|
| 22 |
+
"layer": 3,
|
| 23 |
+
"lm_name": "EleutherAI/pythia-70m-deduped",
|
| 24 |
+
"wandb_name": "SampledActivationTrainer-leverage-EleutherAI/pythia-70m-deduped-resid_post_layer_3_trainer_1",
|
| 25 |
+
"submodule_name": "resid_post_layer_3"
|
| 26 |
+
},
|
| 27 |
+
"buffer": {
|
| 28 |
+
"d_submodule": 512,
|
| 29 |
+
"io": "out",
|
| 30 |
+
"n_ctxs": 244,
|
| 31 |
+
"ctx_len": 1024,
|
| 32 |
+
"refresh_batch_size": 64,
|
| 33 |
+
"out_batch_size": 2048,
|
| 34 |
+
"device": "cuda:0"
|
| 35 |
+
}
|
| 36 |
+
}
|
._test_run_EleutherAI_pythia-70m-deduped_top_k_sampled_sae/resid_post_layer_3/SampledActivationTrainer-leverage-EleutherAI_pythia-70m-deduped-resid_post_layer_3_trainer_1/eval_results.json
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
{"l2_loss": 769.8623352050781, "l1_loss": 14121.056091308594, "l0": 32569.235229492188, "frac_variance_explained": -3766.6612548828125, "cossim": 0.8414181843400002, "l2_ratio": 53.31753587722778, "relative_reconstruction_bias": 62.42244005203247, "loss_original": 2.997953861951828, "loss_reconstructed": 18.36150825023651, "loss_zero": 12.705852329730988, "frac_recovered": -0.5867955405265093, "frac_alive": 1.0, "hyperparameters": {"n_inputs": 200, "context_length": 1024}}
|
._test_run_EleutherAI_pythia-70m-deduped_top_k_sampled_sae/resid_post_layer_3/TopKTrainer-EleutherAI_pythia-70m-deduped-resid_post_layer_3_trainer_0/ae.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f6a919db78dafaa29370b2e7bf06c9e8afe1f3d45cf37d3e51259613f706127e
|
| 3 |
+
size 268701718
|
._test_run_EleutherAI_pythia-70m-deduped_top_k_sampled_sae/resid_post_layer_3/TopKTrainer-EleutherAI_pythia-70m-deduped-resid_post_layer_3_trainer_0/config.json
ADDED
|
@@ -0,0 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"trainer": {
|
| 3 |
+
"trainer_class": "TopKTrainer",
|
| 4 |
+
"dict_class": "AutoEncoderTopK",
|
| 5 |
+
"lr": 0.0003,
|
| 6 |
+
"steps": 48,
|
| 7 |
+
"auxk_alpha": 0.03125,
|
| 8 |
+
"warmup_steps": 1000,
|
| 9 |
+
"decay_start": 38,
|
| 10 |
+
"threshold_beta": 0.999,
|
| 11 |
+
"threshold_start_step": 1000,
|
| 12 |
+
"seed": 0,
|
| 13 |
+
"activation_dim": 512,
|
| 14 |
+
"dict_size": 65536,
|
| 15 |
+
"k": 20,
|
| 16 |
+
"device": "cuda:0",
|
| 17 |
+
"layer": 3,
|
| 18 |
+
"lm_name": "EleutherAI/pythia-70m-deduped",
|
| 19 |
+
"wandb_name": "TopKTrainer-EleutherAI/pythia-70m-deduped-resid_post_layer_3_trainer_0",
|
| 20 |
+
"submodule_name": "resid_post_layer_3"
|
| 21 |
+
},
|
| 22 |
+
"buffer": {
|
| 23 |
+
"d_submodule": 512,
|
| 24 |
+
"io": "out",
|
| 25 |
+
"n_ctxs": 244,
|
| 26 |
+
"ctx_len": 1024,
|
| 27 |
+
"refresh_batch_size": 64,
|
| 28 |
+
"out_batch_size": 2048,
|
| 29 |
+
"device": "cuda:0"
|
| 30 |
+
}
|
| 31 |
+
}
|
._test_run_EleutherAI_pythia-70m-deduped_top_k_sampled_sae/resid_post_layer_3/TopKTrainer-EleutherAI_pythia-70m-deduped-resid_post_layer_3_trainer_0/eval_results.json
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
{"l2_loss": 10.643135368824005, "l1_loss": 39.116480588912964, "l0": 20.0, "frac_variance_explained": 0.2406930886209011, "cossim": 0.7164953276515007, "l2_ratio": 0.9136376194655895, "relative_reconstruction_bias": 1.275299347937107, "loss_original": 2.997953861951828, "loss_reconstructed": 5.598592609167099, "loss_zero": 12.705852329730988, "frac_recovered": 0.732391856610775, "frac_alive": 0.99761962890625, "hyperparameters": {"n_inputs": 200, "context_length": 1024}}
|