andyjzhao commited on
Commit
d4c247f
·
verified ·
1 Parent(s): c6abdf9

Add sanitized release config

Browse files
Files changed (1) hide show
  1. hydra_cfg.yaml +64 -0
hydra_cfg.yaml ADDED
@@ -0,0 +1,64 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ arch: hnet
2
+ bp_per_token: 32
3
+ max_len: 12800
4
+ max_length: 12800
5
+ min_routing_tokens: 8
6
+ r_hi: 0.3
7
+ r_low: 0.0
8
+ r_warm_up_end: 750
9
+ r_warm_up_start: 200
10
+ reference_loss: null
11
+ region_info: promoter1_cds1_utr2_exon2_intron8_nig8_dig16
12
+ strictness_exp: 1.0
13
+ strictness_max: 1.0
14
+ tokenizer: fast
15
+ use_routing_floor: true
16
+ model_cfg:
17
+ arch_layout:
18
+ - m2
19
+ - - m2
20
+ - - m15
21
+ - m2
22
+ - m2
23
+ attn_cfg:
24
+ num_heads:
25
+ - 8
26
+ - 8
27
+ - 12
28
+ rotary_emb_dim:
29
+ - 16
30
+ - 16
31
+ - 24
32
+ window_size:
33
+ - 511
34
+ - 511
35
+ - -1
36
+ d_intermediate:
37
+ - 0
38
+ - 0
39
+ - 2048
40
+ d_model:
41
+ - 512
42
+ - 512
43
+ - 768
44
+ min_routing_tokens: 8
45
+ n_gpt: 1.0
46
+ r_hi: 0.3
47
+ r_low: 0.0
48
+ r_warm_up_end: 750
49
+ r_warm_up_start: 200
50
+ ssm_cfg:
51
+ chunk_size: 256
52
+ d_conv: 4
53
+ d_state: 64
54
+ expand: 2
55
+ head_dim: 64
56
+ tie_embeddings: true
57
+ vocab_size: 12
58
+ hf_user: andyjzhao
59
+ hf_repo: andyjzhao/GeneZip-70M-Transcript-balanced
60
+ model_name: GeneZip-70M-Transcript-balanced
61
+ upload_to_hf: false
62
+ routing_floor:
63
+ k_min_list:
64
+ - 8