| {"seed": 49, "batch_size": 4096, "buffer_mult": 128, "lr": 5e-05, "num_tokens": 400000000, "l1_coeff": 2, "beta1": 0.9, "beta2": 0.999, "d_in": 2304, "dict_size": 16384, "seq_len": 512, "enc_dtype": "float32", "model_name": "gemma-2-2b", "site": "resid_pre", "device_A": "cuda:0", "device_B": "cuda:1", "device_sae": "cuda:2", "dtype": "bfloat16", "model_batch_size": 32, "log_every": 100, "save_every": 30000, "dec_init_norm": 0.08, "hook_point": "blocks.14.hook_resid_pre", "wandb_project": "golden-gate-clip-lora", "wandb_entity": "tompollak", "dump_dir": "./checkpoints"} |