File size: 355 Bytes
d94f6f8 b29a571 7f80564 541c673 b29a571 541c673 b29a571 541c673 0373e8d 88ed85e d94f6f8 b29a571 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
from transformers import PretrainedConfig
class EmotionClassifierConfig(PretrainedConfig):
model_type = "hubert"
def __init__(
self,
hidden_size_lstm=128,
num_classes=6,
**kwargs,
):
super().__init__(**kwargs)
self.hidden_size_lstm = hidden_size_lstm
self.num_classes = num_classes
|