andyjzhao commited on
Commit
b1ab41b
·
verified ·
1 Parent(s): 2825606

Add sanitized release config

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