m97j commited on
Commit
31a0ae3
·
verified ·
1 Parent(s): 3594585

Upload config.json

Browse files
Files changed (1) hide show
  1. config.json +45 -0
config.json ADDED
@@ -0,0 +1,45 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "meta": {
3
+ "model_name": "pose-centric-multimodal-har",
4
+ "version": "v1.0",
5
+ "description": "Pose-centric multimodal temporal HAR model"
6
+ },
7
+ "pose_model": {
8
+ "backbone": "PoseFormerFactorized",
9
+ "joints": 17,
10
+ "in_channels": 3,
11
+ "hidden_dim": 128,
12
+ "layers": 4,
13
+ "heads": 8,
14
+ "max_seq_len": 30,
15
+ "use_sinusoidal_pe": false,
16
+ "enable_temporal_attention": true,
17
+ "return_tokens": true
18
+ },
19
+ "long_term_model": {
20
+ "enabled": true,
21
+ "type": "transformer_encoder",
22
+ "layers": 2,
23
+ "heads": 8,
24
+ "dropout": 0.1
25
+ },
26
+ "image_model": {
27
+ "enabled": true,
28
+ "backbone": "resnet18",
29
+ "feature_dim": 128,
30
+ "pretrained": true
31
+ },
32
+ "fusion_model": {
33
+ "enabled": true,
34
+ "mode": "concat",
35
+ "output_dim": 128
36
+ },
37
+ "classifier": {
38
+ "num_classes": 500
39
+ },
40
+ "regularization": {
41
+ "short_term_dropout": 0.3,
42
+ "long_term_dropout": 0.1
43
+ },
44
+ "precision": "fp32"
45
+ }