KhaldiAbderrhmane commited on
Commit
ea1de2a
·
verified ·
1 Parent(s): 2dc2197

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 is None:
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,