KhaldiAbderrhmane
/

emotion-classifier-hubert

Text Classification

Model card Files Files and versions Community

KhaldiAbderrhmane commited on Jun 4

Commit

470ed02

•

1 Parent(s): 83d1a46

Update config.json

Files changed (1) hide show

config.json +2 -2

config.json CHANGED Viewed

@@ -49,7 +49,7 @@
   "final_dropout": 0.1,
   "hidden_act": "gelu",
   "hidden_dropout": 0.1,
-  "hidden_size": 128,
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "layer_norm_eps": 1e-05,
@@ -61,7 +61,7 @@
   "mask_time_min_masks": 2,
   "mask_time_prob": 0.05,
   "model_type": "hubert",
-  "num_attention_heads": 8,
   "num_classes": 6,
   "num_conv_pos_embedding_groups": 16,
   "num_conv_pos_embeddings": 128,

   "final_dropout": 0.1,
   "hidden_act": "gelu",
   "hidden_dropout": 0.1,
+  "hidden_size_lstm": 128,
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "layer_norm_eps": 1e-05,
   "mask_time_min_masks": 2,
   "mask_time_prob": 0.05,
   "model_type": "hubert",
+  "num_attention_heads": 12,
   "num_classes": 6,
   "num_conv_pos_embedding_groups": 16,
   "num_conv_pos_embeddings": 128,