mhussainahmad commited on
Commit
f86fe7c
·
verified ·
1 Parent(s): b1b8585

Upload config.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. config.json +13 -12
config.json CHANGED
@@ -1,15 +1,16 @@
1
  {
2
  "model_type": "averformer-v4",
3
  "architecture": "AVERFormerV4",
4
- "corpus": "CREMA-D",
5
- "num_classes": 6,
6
  "emotions": [
 
 
 
7
  "anger",
8
- "disgust",
9
  "fear",
10
- "happy",
11
- "neutral",
12
- "sad"
13
  ],
14
  "audio_backbone": "microsoft/wavlm-large",
15
  "video_backbone": "MCG-NJU/videomae-large",
@@ -44,14 +45,14 @@
44
  },
45
  "text_format": {
46
  "tokenizer": "microsoft/deberta-v3-large",
47
- "max_length": 320,
48
- "n_ctx_turns": 5,
49
  "speaker_embedding": true
50
  },
51
  "select_metric": "wF1",
52
- "source_checkpoint": "best.pth",
53
  "training_script": "train_v5_spec.py / train_v6_supcon.py / train_v10_mcncl.py",
54
- "val_score": 0.8320558803202199,
55
- "best_epoch": 12,
56
- "seed": 100
57
  }
 
1
  {
2
  "model_type": "averformer-v4",
3
  "architecture": "AVERFormerV4",
4
+ "corpus": "MELD",
5
+ "num_classes": 7,
6
  "emotions": [
7
+ "neutral",
8
+ "joy",
9
+ "sadness",
10
  "anger",
 
11
  "fear",
12
+ "disgust",
13
+ "surprise"
 
14
  ],
15
  "audio_backbone": "microsoft/wavlm-large",
16
  "video_backbone": "MCG-NJU/videomae-large",
 
45
  },
46
  "text_format": {
47
  "tokenizer": "microsoft/deberta-v3-large",
48
+ "max_length": 256,
49
+ "n_ctx_turns": 3,
50
  "speaker_embedding": true
51
  },
52
  "select_metric": "wF1",
53
+ "source_checkpoint": "pytorch_model.pth",
54
  "training_script": "train_v5_spec.py / train_v6_supcon.py / train_v10_mcncl.py",
55
+ "val_score": 0.3815709082075283,
56
+ "best_epoch": 4,
57
+ "seed": 1337
58
  }