KhaldiAbderrhmane commited on
Commit
88ed85e
·
verified ·
1 Parent(s): 5e4ff4e

Update configuration_emotion_classifier.py

Browse files
Files changed (1) hide show
  1. configuration_emotion_classifier.py +11 -11
configuration_emotion_classifier.py CHANGED
@@ -1,14 +1,14 @@
1
- from transformers import PretrainedConfig
2
 
3
- class EmotionClassifierConfig(PretrainedConfig):
4
-
5
-
6
- def __init__(
7
- self,
8
- hidden_size=128,
9
- num_classes=6,
10
- **kwargs,
11
- ):
12
- super().__init__(**kwargs)
13
  self.hidden_size = hidden_size
14
  self.num_classes = num_classes
 
 
1
+ from transformers import HubertConfig, PretrainedConfig
2
 
3
+ class EmotionClassifierConfig(HubertConfig):
4
+ def __init__(self, hidden_size=128, num_classes=6, **kwargs):
5
+ super().__init__(
6
+ encoder_layers=12, # Default value from HubertConfig
7
+ encoder_attention_heads=12, # Default value from HubertConfig
8
+ encoder_ffn_dim=3072, # Default value from HubertConfig
9
+ # Add other required parameters from HubertConfig here
10
+ **kwargs
11
+ )
 
12
  self.hidden_size = hidden_size
13
  self.num_classes = num_classes
14
+