forwarder1121 commited on
Commit
7a8f73b
·
1 Parent(s): d51ba2e

Upload fine-tuned AST model

Browse files
Files changed (3) hide show
  1. config.json +41 -0
  2. model.safetensors +3 -0
  3. preprocessor_config.json +13 -0
config.json ADDED
@@ -0,0 +1,41 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "MIT/ast-finetuned-audioset-10-10-0.4593",
3
+ "architectures": [
4
+ "ASTForAudioClassification"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.0,
7
+ "frequency_stride": 10,
8
+ "hidden_act": "gelu",
9
+ "hidden_dropout_prob": 0.0,
10
+ "hidden_size": 768,
11
+ "id2label": {
12
+ "0": "ANG",
13
+ "1": "DIS",
14
+ "2": "FEA",
15
+ "3": "HAP",
16
+ "4": "NEU",
17
+ "5": "SAD"
18
+ },
19
+ "initializer_range": 0.02,
20
+ "intermediate_size": 3072,
21
+ "label2id": {
22
+ "ANG": 0,
23
+ "DIS": 1,
24
+ "FEA": 2,
25
+ "HAP": 3,
26
+ "NEU": 4,
27
+ "SAD": 5
28
+ },
29
+ "layer_norm_eps": 1e-12,
30
+ "max_length": 1024,
31
+ "model_type": "audio-spectrogram-transformer",
32
+ "num_attention_heads": 12,
33
+ "num_hidden_layers": 12,
34
+ "num_mel_bins": 128,
35
+ "patch_size": 16,
36
+ "problem_type": "single_label_classification",
37
+ "qkv_bias": true,
38
+ "time_stride": 10,
39
+ "torch_dtype": "float32",
40
+ "transformers_version": "4.46.2"
41
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:66d5ed54d320c0a46f23d97e6db8d7cff693baca7f831b8bb70e8f43d305befe
3
+ size 344802344
preprocessor_config.json ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "do_normalize": true,
3
+ "feature_extractor_type": "ASTFeatureExtractor",
4
+ "feature_size": 1,
5
+ "max_length": 1024,
6
+ "mean": -4.2677393,
7
+ "num_mel_bins": 128,
8
+ "padding_side": "right",
9
+ "padding_value": 0.0,
10
+ "return_attention_mask": false,
11
+ "sampling_rate": 16000,
12
+ "std": 4.5689974
13
+ }