from transformers import HubertConfig, PretrainedConfig | |
class EmotionClassifierConfig(HubertConfig): | |
def __init__(self, hidden_size=128, num_classes=6, **kwargs): | |
super().__init__( | |
encoder_layers=12, # Default value from HubertConfig | |
encoder_attention_heads=12, # Default value from HubertConfig | |
encoder_ffn_dim=3072, # Default value from HubertConfig | |
# Add other required parameters from HubertConfig here | |
**kwargs | |
) | |
self.hidden_size = hidden_size | |
self.num_classes = num_classes | |