KhaldiAbderrhmane commited on
Commit
0373e8d
1 Parent(s): 1d451be

Update configuration_emotion_classifier.py

Browse files
Files changed (1) hide show
  1. configuration_emotion_classifier.py +14 -14
configuration_emotion_classifier.py CHANGED
@@ -1,14 +1,14 @@
1
- from transformers import PretrainedConfig
2
-
3
- class EmotionClassifierConfig(PretrainedConfig):
4
- model_type = "emotion_classifier"
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 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