{"sae": {"expansion_factor": 16, "normalize_decoder": true, "num_latents": 0, "k": 128, "multi_topk": false, "jumprelu": false, "jumprelu_init_threshold": 0.001, "jumprelu_bandwidth": 0.001, "jumprelu_target_l0": null, "jumprelu_per_layer_l0": false, "init_enc_as_dec_transpose": true, "init_b_dec_as_zeros": false}, "batch_size": 4, "max_seq_len": 1024, "num_training_tokens": 1000000000, "cycle_iterator": true, "grad_acc_steps": 1, "micro_acc_steps": 1, "adam_8bit": false, "adam_epsilon": 1e-08, "adam_betas": [0.9, 0.999], "lr": 0.0012, "lr_scheduler_name": "constant", "lr_warmup_steps": 0.0, "l1_coefficient": 0.0, "l1_warmup_steps": 0.0, "use_l2_loss": true, "auxk_alpha": 0.0, "dead_feature_threshold": 10000000, "hookpoints": ["layers.0", "layers.1", "layers.2", "layers.3", "layers.4", "layers.5", "layers.6", "layers.7", "layers.8", "layers.9", "layers.10"], "layers": [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10], "layer_stride": 1, "distribute_modules": false, "save_every": 10000, "normalize_activations": 1, "num_norm_estimation_tokens": 2000000, "clusters": {"k6-c2": [6, 7, 8, 9, 10], "baseline-6": [6], "baseline-7": [7], "baseline-8": [8], "baseline-9": [9], "baseline-10": [10]}, "cluster_hookpoints": {"k6-c2": ["layers.6", "layers.7", "layers.8", "layers.9", "layers.10"], "baseline-6": ["layers.6"], "baseline-7": ["layers.7"], "baseline-8": ["layers.8"], "baseline-9": ["layers.9"], "baseline-10": ["layers.10"]}, "hook": null, "resume_from": null, "keep_last_n": 1, "log_to_wandb": true, "run_name": "checkpoints-clusters/google/gemma-2-2b-1024-topk-128-lambda-0.0-target-L0-None-lr-0.0012", "wandb_log_frequency": 1}