config.json

{"sae": {"expansion_factor": 16, "normalize_decoder": true, "num_latents": 0, "k": 64, "multi_topk": false}, "batch_size": 32, "grad_acc_steps": 1, "micro_acc_steps": 1, "lr": null, "lr_warmup_steps": 1000, "auxk_alpha": 0.0, "dead_feature_threshold": 10000000, "hookpoints": ["layers.8", "layers.10", "layers.12", "layers.14", "layers.16", "layers.18"], "layers": [8, 10, 12, 14, 16, 18], "layer_stride": 1, "distribute_modules": true, "save_every": 500, "log_to_wandb": true, "run_name": "qwen-math", "wandb_log_frequency": 1, "model": "Qwen/Qwen2.5-1.5B-Instruct", "dataset": "huypn16/MetaMath-SAE", "split": "train", "ctx_len": 1024, "hf_token": null, "load_in_8bit": false, "max_examples": null, "resume": false, "seed": 42, "data_preprocessing_num_proc": 112}