Upload config.json with huggingface_hub
Browse files- config.json +13 -12
config.json
CHANGED
|
@@ -1,15 +1,16 @@
|
|
| 1 |
{
|
| 2 |
"model_type": "averformer-v4",
|
| 3 |
"architecture": "AVERFormerV4",
|
| 4 |
-
"corpus": "
|
| 5 |
-
"num_classes":
|
| 6 |
"emotions": [
|
|
|
|
|
|
|
|
|
|
| 7 |
"anger",
|
| 8 |
-
"disgust",
|
| 9 |
"fear",
|
| 10 |
-
"
|
| 11 |
-
"
|
| 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":
|
| 48 |
-
"n_ctx_turns":
|
| 49 |
"speaker_embedding": true
|
| 50 |
},
|
| 51 |
"select_metric": "wF1",
|
| 52 |
-
"source_checkpoint": "
|
| 53 |
"training_script": "train_v5_spec.py / train_v6_supcon.py / train_v10_mcncl.py",
|
| 54 |
-
"val_score": 0.
|
| 55 |
-
"best_epoch":
|
| 56 |
-
"seed":
|
| 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 |
}
|