KhaldiAbderrhmane
commited on
Update configuration_emotion_classifier.py
Browse files
configuration_emotion_classifier.py
CHANGED
@@ -7,14 +7,14 @@ class EmotionClassifierConfig(PretrainedConfig):
|
|
7 |
self,
|
8 |
hidden_size_lstm=128,
|
9 |
num_classes=6,
|
10 |
-
hubert_config=None,
|
11 |
**kwargs,
|
12 |
):
|
13 |
super().__init__(**kwargs)
|
14 |
self.hidden_size_lstm = hidden_size_lstm
|
15 |
self.num_classes = num_classes
|
16 |
|
17 |
-
if hubert_config
|
18 |
# Set default HuBERT configuration
|
19 |
self.hubert_config = HubertConfig(
|
20 |
#num_attention_heads=16,
|
|
|
7 |
self,
|
8 |
hidden_size_lstm=128,
|
9 |
num_classes=6,
|
10 |
+
hubert_config="None",
|
11 |
**kwargs,
|
12 |
):
|
13 |
super().__init__(**kwargs)
|
14 |
self.hidden_size_lstm = hidden_size_lstm
|
15 |
self.num_classes = num_classes
|
16 |
|
17 |
+
if hubert_config == "None":
|
18 |
# Set default HuBERT configuration
|
19 |
self.hubert_config = HubertConfig(
|
20 |
#num_attention_heads=16,
|