light156 commited on
Commit
2fedc53
·
verified ·
1 Parent(s): e479180

Add Sensori model configuration

Browse files
Files changed (1) hide show
  1. config_model.yaml +26 -0
config_model.yaml ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Conv1D configuration
2
+ in_channels: 3
3
+ norm_mode: group_norm
4
+ conv_bias: false
5
+ conv_layer_config: [
6
+ [512, 10, 5],
7
+ [512, 3, 2],
8
+ [512, 3, 2],
9
+ [512, 3, 2],
10
+ [512, 2, 2],
11
+ [512, 2, 2],
12
+ [512, 2, 2],
13
+ ]
14
+
15
+ # Transformer Configuration
16
+ dropout: 0.1
17
+ window_len: 5 # minutes represented by each Transformer token
18
+ conv_pos: 32 # 160 minutes with 5-minute tokens
19
+ conv_pos_groups: 16
20
+ encoder_ffn_embed_dim: 1024
21
+ encoder_attention_heads: 12
22
+ activation_fn: gelu
23
+ encoder_dim: 768
24
+ encoder_layers: 8
25
+ encoder_layerdrop: 0.1
26
+ positional_embedding: relative # absolute or relative