Robzy commited on
Commit
f1ac6d2
·
verified ·
1 Parent(s): bc67d5a

Upload config

Browse files
Files changed (1) hide show
  1. config.json +3 -38
config.json CHANGED
@@ -1,58 +1,23 @@
1
  {
2
- "abs_pos_patch_grid_size": null,
3
  "aum_pretrain": true,
4
- "aum_pretrain_fstride": null,
5
  "aum_pretrain_path": "models/aum-base_audioset-vggsound.pth",
6
- "aum_pretrain_tstride": null,
7
- "bilinear_rope": false,
8
  "bimamba_type": "v1",
9
- "channels": 1,
10
- "depth": 24,
11
- "device": null,
12
- "drop_path_rate": 0,
13
- "drop_rate": 0.0,
14
- "dtype": null,
15
  "embed_dim": 768,
16
- "final_pool_type": "mean",
17
- "flexible_patch_sizes": null,
18
- "flip_img_sequences_ratio": -1.0,
19
- "ft_seq_len": null,
20
- "fused_add_norm": true,
21
- "if_abs_pos_embed": true,
22
- "if_bidirectional": false,
23
- "if_bimamba": false,
24
- "if_cls_token": true,
25
- "if_devide_out": true,
26
- "if_rope": false,
27
- "if_rope_residual": false,
28
- "imagenet_load_double_cls_token": false,
29
- "imagenet_load_middle_cls_token": true,
30
  "imagenet_pretrain": false,
31
- "imagenet_pretrain_modelkey": "model",
32
  "imagenet_pretrain_path": null,
33
- "init_layer_scale": null,
34
- "initializer_cfg": null,
35
- "norm_epsilon": 1e-05,
36
  "num_classes": 309,
37
  "patch_size": [
38
  16,
39
  16
40
  ],
41
- "pt_hw_seq_len": null,
42
- "residual_in_fp32": true,
43
- "rms_norm": true,
44
  "spectrogram_size": [
45
  128,
46
  1024
47
  ],
48
- "ssm_cfg": null,
49
  "strides": [
50
  16,
51
  16
52
  ],
53
- "transpose_token_sequence": false,
54
- "use_PI_for_patch_embed": true,
55
- "use_double_cls_token": false,
56
- "use_end_cls_token": false,
57
- "use_middle_cls_token": true
58
- }
 
1
  {
 
2
  "aum_pretrain": true,
 
3
  "aum_pretrain_path": "models/aum-base_audioset-vggsound.pth",
 
 
4
  "bimamba_type": "v1",
 
 
 
 
 
 
5
  "embed_dim": 768,
 
 
 
 
 
 
 
 
 
 
 
 
 
 
6
  "imagenet_pretrain": false,
 
7
  "imagenet_pretrain_path": null,
8
+ "model_type": "mamba",
 
 
9
  "num_classes": 309,
10
  "patch_size": [
11
  16,
12
  16
13
  ],
 
 
 
14
  "spectrogram_size": [
15
  128,
16
  1024
17
  ],
 
18
  "strides": [
19
  16,
20
  16
21
  ],
22
+ "transformers_version": "4.35.2"
23
+ }