sboughorbel commited on
Commit
1c46cb3
·
verified ·
1 Parent(s): aff666c

Upload gemma-2-9b-it-DPO-L20-k100-lr1e-04-dpo-simpo-Crosscoder dictionary model

Browse files
Files changed (1) hide show
  1. trainer_config.json +35 -0
trainer_config.json ADDED
@@ -0,0 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "trainer": {
3
+ "trainer_class": "BatchTopKCrossCoderTrainer",
4
+ "dict_class": "BatchTopKCrossCoder",
5
+ "lr": 0.0001,
6
+ "steps": 48828,
7
+ "auxk_alpha": 0.03125,
8
+ "warmup_steps": 1000,
9
+ "decay_start": null,
10
+ "threshold_beta": 0.999,
11
+ "threshold_start_step": 1000,
12
+ "top_k_aux": 1792,
13
+ "seed": null,
14
+ "activation_dim": 3584,
15
+ "dict_size": 114688,
16
+ "k": 100,
17
+ "code_normalization": "CROSSCODER",
18
+ "code_normalization_alpha_sae": 1.0,
19
+ "code_normalization_alpha_cc": 0.1,
20
+ "device": "cuda",
21
+ "layer": 20,
22
+ "lm_name": "princeton-nlp/gemma-2-9b-it-SimPO-princeton-nlp/gemma-2-9b-it-DPO",
23
+ "wandb_name": "gemma-2-9b-it-DPO-L20-k100-lr1e-04-dpo-simpo-Crosscoder",
24
+ "submodule_name": null,
25
+ "dict_class_kwargs": {
26
+ "same_init_for_all_layers": false,
27
+ "norm_init_scale": 0.005,
28
+ "init_with_transpose": false,
29
+ "encoder_layers": null,
30
+ "code_normalization": "crosscoder",
31
+ "code_normalization_alpha_sae": 1.0,
32
+ "code_normalization_alpha_cc": 0.1
33
+ }
34
+ }
35
+ }